Mila::Dnn::Compute::CpuResidualOp Member List

This is the complete list of members for Mila::Dnn::Compute::CpuResidualOp, including all inherited members.

asLeftTensor(const ITensor &t)	Mila::Dnn::Compute::BinaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	inlineprotectedstatic
asOutputTensor(ITensor &t)	Mila::Dnn::Compute::BinaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	inlineprotectedstatic
asRightTensor(const ITensor &t)	Mila::Dnn::Compute::BinaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	inlineprotectedstatic
backward(const ITensor &input_a, const ITensor &input_b, const ITensor &output_grad, ITensor &input_a_grad, ITensor &input_b_grad) const override	Mila::Dnn::Compute::CpuResidualOp	inlinevirtual
build(const BuildContext &build_context)	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	inlinevirtual
clearGradients() noexcept	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	inlinevirtual
config_	Mila::Dnn::Compute::CpuResidualOp	private
context_	Mila::Dnn::Compute::CpuResidualOp	private
CpuResidualOp(IExecutionContext *context, const ResidualConfig &config)	Mila::Dnn::Compute::CpuResidualOp	inline
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 &input_a, const ITensor &input_b, ITensor &output) const override	Mila::Dnn::Compute::CpuResidualOp	inlinevirtual
getDataType() const	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	inlinevirtual
getDeviceType() const	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	inlinevirtual
getName() const override	Mila::Dnn::Compute::CpuResidualOp	inlinevirtual
getOperationType() const override	Mila::Dnn::Compute::CpuResidualOp	inlinevirtual
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::CpuResidualOp
ParameterGradTensor typedef	Mila::Dnn::Compute::BinaryOperation< DeviceType::Cpu, TensorDataType::FP32 >
ParameterTensor typedef	Mila::Dnn::Compute::BinaryOperation< DeviceType::Cpu, TensorDataType::FP32 >
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
TensorLeftType typedef	Mila::Dnn::Compute::BinaryOperation< DeviceType::Cpu, TensorDataType::FP32 >
TensorOutputType typedef	Mila::Dnn::Compute::BinaryOperation< DeviceType::Cpu, TensorDataType::FP32 >
TensorRightType typedef	Mila::Dnn::Compute::BinaryOperation< DeviceType::Cpu, TensorDataType::FP32 >
training_mode_	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	protected
~BinaryOperation()=default	Mila::Dnn::Compute::BinaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	virtual
~Operation()=default	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	virtual