Mila
Deep Neural Network Library
|
This is the complete list of members for Mila::Dnn::SoftmaxConfig, including all inherited members.
axis_ | Mila::Dnn::SoftmaxConfig | private |
getAxis() const | Mila::Dnn::SoftmaxConfig | inline |
getName() const | Mila::Dnn::ComponentConfig | inline |
getPrecision() const | Mila::Dnn::ComponentConfig | inline |
is_training_ | Mila::Dnn::ComponentConfig | protected |
isTraining() const | Mila::Dnn::ComponentConfig | inline |
name_ | Mila::Dnn::ComponentConfig | protected |
precision_ | Mila::Dnn::ComponentConfig | protected |
SoftmaxConfig()=default | Mila::Dnn::SoftmaxConfig | |
validate() const | Mila::Dnn::SoftmaxConfig | inlinevirtual |
withAxis(int64_t axis) | Mila::Dnn::SoftmaxConfig | inline |
withName(this Self &&self, std::string name) | Mila::Dnn::ComponentConfig | inline |
withPrecision(this Self &&self, ComputePrecision::Policy policy) | Mila::Dnn::ComponentConfig | inline |
withTraining(this Self &&self, bool is_training) | Mila::Dnn::ComponentConfig | inline |
~ComponentConfig()=default | Mila::Dnn::ComponentConfig | virtual |