Mila::Dnn::Compute::CpuAttentionOp Member List

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

allocateStateTensors()	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
applySoftmax() const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
asInputTensor(const ITensor &t)	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	inlineprotectedstatic
asOutputTensor(ITensor &t)	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	inlineprotectedstatic
att_	Mila::Dnn::Compute::CpuAttentionOp	private
att_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
B_	Mila::Dnn::Compute::CpuAttentionOp	private
backward(const ITensor &input, const ITensor &output_grad, ITensor &input_grad) const override	Mila::Dnn::Compute::CpuAttentionOp	inlinevirtual
build(const BuildContext &config) override	Mila::Dnn::Compute::CpuAttentionOp	inlinevirtual
clearGradients() noexcept	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
computeAttentionScores(float scale) const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
computeGradientAtt() const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
computeGradientK() const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
computeGradientPreatt(float scale) const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
computeGradientQ() const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
computeGradientV() const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
computeOutputValues() const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
config_	Mila::Dnn::Compute::CpuAttentionOp	private
context_	Mila::Dnn::Compute::CpuAttentionOp	private
CpuAttentionOp(IExecutionContext *context, const MultiHeadAttentionConfig &config)	Mila::Dnn::Compute::CpuAttentionOp	inlineexplicit
CpuExecutionContext typedef	Mila::Dnn::Compute::CpuAttentionOp
data_type	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	static
DataTypeTraits typedef	Mila::Dnn::Compute::Operation< TDeviceType, TInput >
datt_	Mila::Dnn::Compute::CpuAttentionOp	private
datt_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
device_type	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	static
dk_	Mila::Dnn::Compute::CpuAttentionOp	private
dk_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
dpreatt_	Mila::Dnn::Compute::CpuAttentionOp	private
dpreatt_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
dq_	Mila::Dnn::Compute::CpuAttentionOp	private
dq_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
dv_	Mila::Dnn::Compute::CpuAttentionOp	private
dv_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
dvout_	Mila::Dnn::Compute::CpuAttentionOp	private
dvout_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
embedding_dim_	Mila::Dnn::Compute::CpuAttentionOp	private
forward(const ITensor &input, ITensor &output) const override	Mila::Dnn::Compute::CpuAttentionOp	inlinevirtual
getDataType() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
getDeviceType() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
getName() const override	Mila::Dnn::Compute::CpuAttentionOp	inlinevirtual
getOperationType() const override	Mila::Dnn::Compute::CpuAttentionOp	inlinevirtual
getStateMemorySize() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
HS_	Mila::Dnn::Compute::CpuAttentionOp	private
is_built_	Mila::Dnn::Compute::CpuAttentionOp	private
isBuilt() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
isEvalMode() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
k_	Mila::Dnn::Compute::CpuAttentionOp	private
k_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
MR typedef	Mila::Dnn::Compute::CpuAttentionOp
NH_	Mila::Dnn::Compute::CpuAttentionOp	private
permute_backward(float *dX) const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
permuteQKV(const float *X) const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
preatt_	Mila::Dnn::Compute::CpuAttentionOp	private
preatt_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
q_	Mila::Dnn::Compute::CpuAttentionOp	private
q_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
qkv_dim_	Mila::Dnn::Compute::CpuAttentionOp	private
setGradients(ITensor *, ITensor *) override	Mila::Dnn::Compute::CpuAttentionOp	inlinevirtual
setParameters(ITensor *, ITensor *) override	Mila::Dnn::Compute::CpuAttentionOp	inlinevirtual
setTrainingMode(TrainingMode training_mode)	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
T_	Mila::Dnn::Compute::CpuAttentionOp	private
TensorInputType typedef	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >
TensorOutputType typedef	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >
TensorType typedef	Mila::Dnn::Compute::CpuAttentionOp
training_mode_	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	protected
unpermute(float *Y) const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
unpermute_backward(const float *dY) const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
v_	Mila::Dnn::Compute::CpuAttentionOp	private
v_out_	Mila::Dnn::Compute::CpuAttentionOp	private
v_out_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
v_tensor_	Mila::Dnn::Compute::CpuAttentionOp	private
validateInputShape(const shape_t &input_shape) const	Mila::Dnn::Compute::CpuAttentionOp	inlineprivate
~CpuAttentionOp() override=default	Mila::Dnn::Compute::CpuAttentionOp
~Operation()=default	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	virtual
~UnaryOperation()=default	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	virtual