Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision > Member List

This is the complete list of members for Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >, including all inherited members.

decode(const TensorType &input, dim_t position)=0	Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >	pure virtual
MR typedef (defined in Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >)	Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >
prefill(const TensorType &input, dim_t position_offset)=0	Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >	pure virtual
resetKvCache()=0	Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >	pure virtual
rewindKvCache(dim_t position)=0	Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >	pure virtual
setState(const GqaState &state)=0	Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >	pure virtual
supportsKvCache() const noexcept=0	Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >	pure virtual
TensorType typedef (defined in Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >)	Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >
~ITransformerBlock()=default (defined in Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >)	Mila::Dnn::ITransformerBlock< TDeviceType, TPrecision >	virtual