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

dropout_	Mila::Dnn::MultiHeadAttentionConfig	private
embedding_dim_	Mila::Dnn::MultiHeadAttentionConfig	private
getDropout() const	Mila::Dnn::MultiHeadAttentionConfig	inline
getEmbeddingDim() const	Mila::Dnn::MultiHeadAttentionConfig	inline
getInputShape() const	Mila::Dnn::MultiHeadAttentionConfig	inline
getName() const	Mila::Dnn::ComponentConfig	inline
getNumHeads() const	Mila::Dnn::MultiHeadAttentionConfig	inline
getPrecision() const	Mila::Dnn::ComponentConfig	inline
getScaleFactor() const	Mila::Dnn::MultiHeadAttentionConfig	inline
input_shape_	Mila::Dnn::MultiHeadAttentionConfig	private
is_training_	Mila::Dnn::ComponentConfig	protected
isTraining() const	Mila::Dnn::ComponentConfig	inline
MultiHeadAttentionConfig(size_t embedding_dim, size_t num_heads)	Mila::Dnn::MultiHeadAttentionConfig	inline
name_	Mila::Dnn::ComponentConfig	protected
num_heads_	Mila::Dnn::MultiHeadAttentionConfig	private
precision_	Mila::Dnn::ComponentConfig	protected
scale_factor_	Mila::Dnn::MultiHeadAttentionConfig	private
separate_projections_	Mila::Dnn::MultiHeadAttentionConfig	private
use_causal_mask_	Mila::Dnn::MultiHeadAttentionConfig	private
useCausalMask() const	Mila::Dnn::MultiHeadAttentionConfig	inline
useSeparateProjections() const	Mila::Dnn::MultiHeadAttentionConfig	inline
validate() const	Mila::Dnn::MultiHeadAttentionConfig	inlinevirtual
withCausalMask(bool causal)	Mila::Dnn::MultiHeadAttentionConfig	inline
withDropout(float dropout)	Mila::Dnn::MultiHeadAttentionConfig	inline
withInputShape(const std::vector< size_t > &input_shape)	Mila::Dnn::MultiHeadAttentionConfig	inline
withName(this Self &&self, std::string name)	Mila::Dnn::ComponentConfig	inline
withPrecision(this Self &&self, ComputePrecision::Policy policy)	Mila::Dnn::ComponentConfig	inline
withScaleFactor(float scale_factor)	Mila::Dnn::MultiHeadAttentionConfig	inline
withSeparateProjections(bool separate_projections)	Mila::Dnn::MultiHeadAttentionConfig	inline
withTraining(this Self &&self, bool is_training)	Mila::Dnn::ComponentConfig	inline
~ComponentConfig()=default	Mila::Dnn::ComponentConfig	virtual