| addTensors(const Tensor< TInput, MR > &a, const Tensor< TInput, MR > &b, Tensor< TInput, MR > &result) | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlineprivate |
| backward(const Tensor< TInput, MR > &input, const Tensor< TOutput, MR > &output_grad, Tensor< TInput, MR > &input_grad) | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inline |
| config_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| connectionTypeToString(ResidualConfig::ConnectionType type) | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlineprivatestatic |
| createContext(const std::string &device_name) | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprivatestatic |
| createOperation() | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlineprivate |
| createProjection(const std::vector< size_t > &input_shape, const std::vector< size_t > &output_shape) | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlineprivate |
| device_context_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
| forward(const Tensor< TInput, MR > &input, Tensor< TOutput, MR > &output) | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inline |
| gate_weights_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| gated_operation_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| getDeviceContext() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| getDeviceType() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| getInnerModule() | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inline |
| getName() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | 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 |
| initializeGateParameters(const std::vector< size_t > &shape) | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlineprivate |
| inner_input_grad_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| inner_module_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| inner_output_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| inner_parameter_grads_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| isTraining() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
| load(ModelArchive &archive) override | Mila::Dnn::Residual< 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 typedef | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | |
| MR typedef | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | |
| operation_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| output_state_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| parameter_grads_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| parameter_map_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | protected |
| parameterCount() const override | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlinevirtual |
| parameters_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| parametersToString() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprotected |
| projection_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| projection_output_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| properties_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| Residual(const std::string &device_name, const ResidualConfig &config) | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlineexplicit |
| Residual(std::shared_ptr< DeviceContext > device_context, const ResidualConfig &config) | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlineexplicit |
| save(ModelArchive &zip) const override | Mila::Dnn::Residual< 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() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprotected |
| temp_grad_ | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | private |
| tensorShapesMatch(const Tensor< TInput, MR > &a, const Tensor< TOutput, MR > &b) | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlineprivate |
| toString() const override | Mila::Dnn::Residual< TDeviceType, TInput, TOutput > | inlinevirtual |
| training_mode_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
| ~Module()=default | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | virtual |
1.9.8