Mila::Dnn::Compute::CpuEncoderOp Member List

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

asInputTensor(const ITensor &t)	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::INT32, TensorDataType::FP32 >	inlineprotectedstatic
asOutputTensor(ITensor &t)	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::INT32, TensorDataType::FP32 >	inlineprotectedstatic
backward(const ITensor &input, const ITensor &output_grad, ITensor &input_grad) const override	Mila::Dnn::Compute::CpuEncoderOp	inlinevirtual
batch_size_	Mila::Dnn::Compute::CpuEncoderOp	private
build(const BuildContext &config) override	Mila::Dnn::Compute::CpuEncoderOp	inlinevirtual
clearGradients() noexcept	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	inlinevirtual
config_	Mila::Dnn::Compute::CpuEncoderOp	private
ConfigType typedef	Mila::Dnn::Compute::CpuEncoderOp
context_	Mila::Dnn::Compute::CpuEncoderOp	private
CpuEncoderOp(IExecutionContext *context, const LpeConfig &config)	Mila::Dnn::Compute::CpuEncoderOp	inlineexplicit
CpuExecutionContext typedef	Mila::Dnn::Compute::CpuEncoderOp
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
embedding_dim_	Mila::Dnn::Compute::CpuEncoderOp	private
forward(const ITensor &input, ITensor &output) const override	Mila::Dnn::Compute::CpuEncoderOp	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::CpuEncoderOp	inlinevirtual
getOperationType() const override	Mila::Dnn::Compute::CpuEncoderOp	inlinevirtual
getStateMemorySize() const	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	inlinevirtual
is_built_	Mila::Dnn::Compute::CpuEncoderOp	private
isBuilt() const	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	inlinevirtual
isEvalMode() const	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	inlinevirtual
MR typedef	Mila::Dnn::Compute::CpuEncoderOp
OperationBase typedef	Mila::Dnn::Compute::CpuEncoderOp
seq_length_	Mila::Dnn::Compute::CpuEncoderOp	private
setGradients(ITensor *wte_grad, ITensor *wpe_grad) override	Mila::Dnn::Compute::CpuEncoderOp	inlinevirtual
setParameters(ITensor *wte, ITensor *wpe) override	Mila::Dnn::Compute::CpuEncoderOp	inlinevirtual
setTrainingMode(TrainingMode training_mode)	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	inlinevirtual
TensorInputType typedef	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::INT32, TensorDataType::FP32 >
TensorOutputType typedef	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::INT32, TensorDataType::FP32 >
TensorType typedef	Mila::Dnn::Compute::CpuEncoderOp
training_mode_	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	protected
validateInputShape(const ITensor &input) const	Mila::Dnn::Compute::CpuEncoderOp	inlineprivate
validateInputShape(const shape_t &input_shape) const	Mila::Dnn::Compute::CpuEncoderOp	inlineprivate
wpe_	Mila::Dnn::Compute::CpuEncoderOp	private
wpe_grad_	Mila::Dnn::Compute::CpuEncoderOp	private
wte_	Mila::Dnn::Compute::CpuEncoderOp	private
wte_grad_	Mila::Dnn::Compute::CpuEncoderOp	private
~CpuEncoderOp() override=default	Mila::Dnn::Compute::CpuEncoderOp
~Operation()=default	Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >	virtual
~UnaryOperation()=default	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::INT32, TensorDataType::FP32 >	virtual