Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::LayerNormConfig Member List

This is the complete list of members for Mila::Dnn::LayerNormConfig, including all inherited members.

axis_Mila::Dnn::LayerNormConfigprivate
epsilon_Mila::Dnn::LayerNormConfigprivate
getAxis() constMila::Dnn::LayerNormConfiginline
getEpsilon() constMila::Dnn::LayerNormConfiginline
getInputShape() constMila::Dnn::LayerNormConfiginline
getName() constMila::Dnn::ComponentConfiginline
getPrecision() constMila::Dnn::ComponentConfiginline
has_bias_Mila::Dnn::LayerNormConfigprivate
hasBias() constMila::Dnn::LayerNormConfiginline
input_shape_Mila::Dnn::LayerNormConfigprivate
is_training_Mila::Dnn::ComponentConfigprotected
isTraining() constMila::Dnn::ComponentConfiginline
LayerNormConfig()=defaultMila::Dnn::LayerNormConfig
LayerNormConfig(size_t normalized_dim)Mila::Dnn::LayerNormConfiginlineexplicit
name_Mila::Dnn::ComponentConfigprotected
precision_Mila::Dnn::ComponentConfigprotected
validate() constMila::Dnn::LayerNormConfiginlinevirtual
withAxis(int64_t axis)Mila::Dnn::LayerNormConfiginline
withBias(bool has_bias)Mila::Dnn::LayerNormConfiginline
withEpsilon(float epsilon)Mila::Dnn::LayerNormConfiginline
withInputShape(const std::vector< size_t > &input_shape)Mila::Dnn::LayerNormConfiginline
withName(this Self &&self, std::string name)Mila::Dnn::ComponentConfiginline
withPrecision(this Self &&self, ComputePrecision::Policy policy)Mila::Dnn::ComponentConfiginline
withTraining(this Self &&self, bool is_training)Mila::Dnn::ComponentConfiginline
~ComponentConfig()=defaultMila::Dnn::ComponentConfigvirtual