Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision > Member List

This is the complete list of members for Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >, including all inherited members.

build(const BuildContext &build_context)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
clearGradients() noexcept	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
data_type	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	static
DataTypeTraits typedef	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
device_type	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	static
getDataType() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
getDeviceType() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
getName() const =0	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	pure virtual
getOperationType() const =0	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	pure virtual
getStateMemorySize() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
is_built_	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	protected
isBuilt() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
isEvalMode() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
setGradients(ITensor *weight_grad, ITensor *bias_grad)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
setParameters(ITensor *weight, ITensor *bias)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
setTrainingMode(TrainingMode training_mode)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
training_mode_	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	protected
~Operation()=default	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	virtual