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

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

axis_Mila::Dnn::SoftmaxConfigprivate
getAxis() constMila::Dnn::SoftmaxConfiginline
getName() constMila::Dnn::ComponentConfiginline
getPrecision() constMila::Dnn::ComponentConfiginline
is_training_Mila::Dnn::ComponentConfigprotected
isTraining() constMila::Dnn::ComponentConfiginline
name_Mila::Dnn::ComponentConfigprotected
precision_Mila::Dnn::ComponentConfigprotected
SoftmaxConfig()=defaultMila::Dnn::SoftmaxConfig
validate() constMila::Dnn::SoftmaxConfiginlinevirtual
withAxis(int64_t axis)Mila::Dnn::SoftmaxConfiginline
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