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

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

class_weights_Mila::Dnn::CrossEntropyConfigprivate
CrossEntropyConfig(int64_t vocab_size)Mila::Dnn::CrossEntropyConfiginlineexplicit
getClassWeights() constMila::Dnn::CrossEntropyConfiginline
getLabelSmoothing() constMila::Dnn::CrossEntropyConfiginline
getName() constMila::Dnn::ComponentConfiginline
getPaddingIndex() constMila::Dnn::CrossEntropyConfiginline
getPrecision() constMila::Dnn::ComponentConfiginline
getReduction() constMila::Dnn::CrossEntropyConfiginline
getVocabSize() constMila::Dnn::CrossEntropyConfiginline
ignore_padding_Mila::Dnn::CrossEntropyConfigprivate
ignorePadding() constMila::Dnn::CrossEntropyConfiginline
is_training_Mila::Dnn::ComponentConfigprotected
isTraining() constMila::Dnn::ComponentConfiginline
label_smoothing_Mila::Dnn::CrossEntropyConfigprivate
name_Mila::Dnn::ComponentConfigprotected
padding_idx_Mila::Dnn::CrossEntropyConfigprivate
precision_Mila::Dnn::ComponentConfigprotected
reduce_Mila::Dnn::CrossEntropyConfigprivate
validate() constMila::Dnn::CrossEntropyConfiginlinevirtual
vocab_size_Mila::Dnn::CrossEntropyConfigprivate
withClassWeights(const std::vector< float > &weights)Mila::Dnn::CrossEntropyConfiginline
withIgnorePadding(bool ignore_pad)Mila::Dnn::CrossEntropyConfiginline
withLabelSmoothing(float smoothing)Mila::Dnn::CrossEntropyConfiginline
withName(this Self &&self, std::string name)Mila::Dnn::ComponentConfiginline
withPaddingIndex(int64_t pad_idx)Mila::Dnn::CrossEntropyConfiginline
withPrecision(this Self &&self, ComputePrecision::Policy policy)Mila::Dnn::ComponentConfiginline
withReduction(bool reduce)Mila::Dnn::CrossEntropyConfiginline
withTraining(this Self &&self, bool is_training)Mila::Dnn::ComponentConfiginline
~ComponentConfig()=defaultMila::Dnn::ComponentConfigvirtual