|
Mila
Deep Neural Network Library
|
This is the complete list of members for Mila::Dnn::Compute::CudaResidualOp< TInput, TOutput >, including all inherited members.
| backward(const Tensor< TInput, MR > &input1, const Tensor< TInput, MR > &input2, const Tensor< TOutput, MR > &output, const Tensor< TOutput, MR > &output_gradient, const std::vector< std::shared_ptr< Tensor< TInput, MR > > > ¶meters, std::vector< std::shared_ptr< Tensor< TOutput, MR > > > ¶meter_gradients, Tensor< TInput, MR > &input1_gradient, Tensor< TInput, MR > &input2_gradient, const OperationAttributes &properties, const std::vector< std::shared_ptr< Tensor< TOutput, MR > > > &output_state) const | Mila::Dnn::Compute::CudaResidualOp< TInput, TOutput > | inline |
| Mila::Dnn::Compute::BinaryOperation::backward(const Tensor< TInput1, MR > &input1, const Tensor< TInput2, MR > &input2, const Tensor< TOutput, MR > &output, const Tensor< TOutput, MR > &output_gradient, const std::vector< std::shared_ptr< Tensor< TInput1, MR > > > ¶meters, std::vector< std::shared_ptr< Tensor< TOutput, MR > > > ¶meter_gradients, Tensor< TInput1, MR > &input1_gradient, Tensor< TInput2, MR > &input2_gradient, const OperationAttributes &attributes, const std::vector< std::shared_ptr< Tensor< TOutput, MR > > > &output_state) const | Mila::Dnn::Compute::BinaryOperation< TDeviceType, TInput1, TInput2, TOutput > | inlinevirtual |
| BinaryOperation(OperationType operation_type) | Mila::Dnn::Compute::BinaryOperation< TDeviceType, TInput1, TInput2, TOutput > | inline |
| BinaryOperation(OperationType operation_type, std::shared_ptr< DeviceContext > context) | Mila::Dnn::Compute::BinaryOperation< TDeviceType, TInput1, TInput2, TOutput > | inline |
| BinaryOperationBase typedef | Mila::Dnn::Compute::CudaResidualOp< TInput, TOutput > | |
| config_ | Mila::Dnn::Compute::CudaResidualOp< TInput, TOutput > | private |
| CudaResidualOp(const ResidualConfig &config) | Mila::Dnn::Compute::CudaResidualOp< TInput, TOutput > | inline |
| CudaResidualOp(std::shared_ptr< DeviceContext > context, const ResidualConfig &config) | Mila::Dnn::Compute::CudaResidualOp< TInput, TOutput > | inline |
| device_context_ | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | private |
| forward(const Tensor< TInput, MR > &input1, const Tensor< TInput, MR > &input2, const std::vector< std::shared_ptr< Tensor< TInput, MR > > > ¶meters, const OperationAttributes &properties, Tensor< TOutput, MR > &output, std::vector< std::shared_ptr< Tensor< TOutput, MR > > > &output_state) const override | Mila::Dnn::Compute::CudaResidualOp< TInput, TOutput > | inline |
| Mila::Dnn::Compute::BinaryOperation::forward(const Tensor< TInput1, MR > &input1, const Tensor< TInput2, MR > &input2, const std::vector< std::shared_ptr< Tensor< TInput1, MR > > > ¶meters, const OperationAttributes &attributes, Tensor< TOutput, MR > &output, std::vector< std::shared_ptr< Tensor< TOutput, MR > > > &output_state) const =0 | Mila::Dnn::Compute::BinaryOperation< TDeviceType, TInput1, TInput2, TOutput > | pure virtual |
| getDeviceContext() const | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | inline |
| getDeviceType() const | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | inline |
| getName() const override | Mila::Dnn::Compute::CudaResidualOp< TInput, TOutput > | inlinevirtual |
| getOperationType() const | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | inline |
| MR typedef | Mila::Dnn::Compute::CudaResidualOp< TInput, TOutput > | |
| operation_type_ | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | private |
| OperationBase(OperationType operation_type, std::shared_ptr< DeviceContext > context) | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | inline |
| ~BinaryOperation()=default | Mila::Dnn::Compute::BinaryOperation< TDeviceType, TInput1, TInput2, TOutput > | virtual |
| ~OperationBase()=default | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | virtual |