| asInputA(const ITensor &t) | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | inlineprotectedstatic |
| asInputB(const ITensor &t) | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | inlineprotectedstatic |
| asOutputTensor(ITensor &t) | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | inlineprotectedstatic |
| backward(const ITensor &grad_out_a, const ITensor &grad_out_b, ITensor &grad_in_a, ITensor &grad_in_b) const =0 | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | pure virtual |
| build(const BuildContext &build_context) | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| clearGradients() noexcept | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| data_type | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | static |
| DataTypeTraits typedef | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | |
| device_type | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | static |
| forward(const ITensor &in_a, const ITensor &in_b, ITensor &out_a, ITensor &out_b) const =0 | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | pure virtual |
| getDataType() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| getDeviceType() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| getName() const=0 | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | pure virtual |
| getOperationType() const=0 | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | pure virtual |
| getStateMemorySize() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| is_built_ | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | protected |
| isBuilt() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| isEvalMode() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| MR typedef | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | |
| setGradients(ITensor *weight_grad, ITensor *bias_grad) | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| setParameters(ITensor *weight, ITensor *bias) | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| setTrainingMode(TrainingMode training_mode) | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| TensorInputAType typedef | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | |
| TensorInputBType typedef | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | |
| TensorOutputType typedef | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | |
| training_mode_ | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | protected |
| ~Operation()=default | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | virtual |
| ~PairedOperation()=default | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > | virtual |
1.15.0