Loading...
Searching...
No Matches
layer_normalization_operator.h File Reference

Go to the source code of this file.

Classes

struct  opennn::LayerNormalizationOperator
 

Namespaces

namespace  opennn
 

Enumerations

enum class  opennn::NormalizationMethod { opennn::LayerNorm , opennn::RMS }
 

Functions

void opennn::layer_normalization_forward (const TensorView &, const TensorView &, const TensorView &, TensorView &, TensorView &, TensorView &, TensorView &, float)
 
void opennn::layer_normalization_add_forward (const TensorView &, const TensorView &, const TensorView &, const TensorView &, TensorView &, TensorView &, TensorView &, TensorView &, TensorView &, float)
 
void opennn::layer_normalization_backward (const TensorView &, const TensorView &, const TensorView &, const TensorView &, const TensorView &, const TensorView &, const TensorView &, const TensorView &, TensorView &, TensorView *=nullptr)
 
void opennn::rms_normalization_forward (const TensorView &, const TensorView &, TensorView &, TensorView &, TensorView &, float)
 
void opennn::rms_normalization_backward (const TensorView &, const TensorView &, const TensorView &, const TensorView &, const TensorView &, const TensorView &, TensorView &)