Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Encoder< TDeviceType, TInput, TOutput > Member List

This is the complete list of members for Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >, including all inherited members.

attributes_Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >private
config_Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >private
createContext(const std::string &device_name)Mila::Dnn::Module< TDeviceType, TInput, TOutput >inlineprivatestatic
createOperation()Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >inlineprivate
device_context_Mila::Dnn::Module< TDeviceType, TInput, TOutput >private
Encoder(const std::string &device_name, const EncoderConfig &config)Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >inlineexplicit
Encoder(std::shared_ptr< DeviceContext > device_context, const EncoderConfig &config)Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >inlineexplicit
forward(const Tensor< TInput, MR > &input, Tensor< TOutput, MR > &output)Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >inline
getChannels() constMila::Dnn::Encoder< TDeviceType, TInput, TOutput >inline
getDeviceContext() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getDeviceType() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getMaxSequenceLength() constMila::Dnn::Encoder< TDeviceType, TInput, TOutput >inline
getName() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getParameterTensors() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getPrecision() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getStateTensors() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getVocabularyLength() constMila::Dnn::Encoder< TDeviceType, TInput, TOutput >inline
initializeTensors()Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >inlineprivate
isTraining() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
load(ModelArchive &archive) overrideMila::Dnn::Encoder< TDeviceType, TInput, TOutput >inlinevirtual
Module(const std::string &device_name, const ComponentConfig &config)Mila::Dnn::Module< TDeviceType, TInput, TOutput >inlineexplicit
Module(std::shared_ptr< DeviceContext > context, const ComponentConfig &config)Mila::Dnn::Module< TDeviceType, TInput, TOutput >inlineexplicit
ModuleBase typedefMila::Dnn::Encoder< TDeviceType, TInput, TOutput >
MR typedefMila::Dnn::Encoder< TDeviceType, TInput, TOutput >
operation_Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >private
output_state_Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >private
parameter_map_Mila::Dnn::Module< TDeviceType, TInput, TOutput >protected
parameterCount() const overrideMila::Dnn::Encoder< TDeviceType, TInput, TOutput >inlinevirtual
parameters_Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >private
parametersToString() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inlineprotected
save(ModelArchive &archive) const overrideMila::Dnn::Encoder< TDeviceType, TInput, TOutput >inlinevirtual
setTraining(bool is_training)Mila::Dnn::Module< TDeviceType, TInput, TOutput >inlinevirtual
state_map_Mila::Dnn::Module< TDeviceType, TInput, TOutput >protected
stateToString() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inlineprotected
toString() const overrideMila::Dnn::Encoder< TDeviceType, TInput, TOutput >inlinevirtual
training_mode_Mila::Dnn::Module< TDeviceType, TInput, TOutput >private
wpe_Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >private
wte_Mila::Dnn::Encoder< TDeviceType, TInput, TOutput >private
~Module()=defaultMila::Dnn::Module< TDeviceType, TInput, TOutput >virtual