| addModule(const std::string &name, std::shared_ptr< Module< TDeviceType, TDataType, TDataType > > module) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
| addModule(std::shared_ptr< Module< TDeviceType, TDataType, TDataType > > module) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
| attn_block_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| attn_output_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| child_module_map_ | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | private |
| child_modules_ | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | private |
| CompositeModule() | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
| CompositeModule(const std::string &device_name, const ComponentConfig &config) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inlineexplicit |
| CompositeModule(std::shared_ptr< DeviceContext > context, const ComponentConfig &config) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inlineexplicit |
| CompositeModuleBase typedef | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | |
| config_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| createContext(const std::string &device_name) | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprivatestatic |
| device_context_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
| dropout_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| forward(const Tensor< TDataType, MR > &input, Tensor< TDataType, MR > &output) | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | inline |
| getDeviceContext() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| getDeviceType() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| getModule(const std::string &name) const | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
| getModules() const | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
| getName() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| getNamedModules() const | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
| getParameterTensors() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| getPrecision() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| getStateTensors() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| hasModule(const std::string &name) const | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
| initializeModules() | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | inlineprivate |
| isTraining() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| ln_1_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| ln_1_output_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| ln_2_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| ln_2_output_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| load(ModelArchive &archive) override | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | inlinevirtual |
| mlp_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| mlp_output_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| 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 typedef | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | |
| MR typedef | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | |
| parameter_map_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | protected |
| parameterCount() const override | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | inlinevirtual |
| parametersToString() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprotected |
| removeModule(const std::string &name) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
| replaceModule(const std::string &name, std::shared_ptr< Module< TDeviceType, TDataType, TDataType > > module) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
| res_1_output_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| res_2_output_ | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | private |
| save(ModelArchive &archive) const override | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | inlinevirtual |
| setTraining(bool is_training) override | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inlinevirtual |
| state_map_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | protected |
| stateToString() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprotected |
| toString() const override | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | inlinevirtual |
| training_mode_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
| TransformerBlock(const std::string &device_name, const TransformerBlockConfig &config) | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | inlineexplicit |
| TransformerBlock(std::shared_ptr< DeviceContext > device_context, const TransformerBlockConfig &config) | Mila::Dnn::TransformerBlock< TDeviceType, TDataType > | inlineexplicit |
| ~CompositeModule()=default | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | virtual |
| ~Module()=default | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | virtual |
1.9.8