Mila::Dnn::GqaConfig Member List

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

fromMetadata(const SerializationMetadata &meta)	Mila::Dnn::GqaConfig	inlinevirtual
getGroupSize() const noexcept	Mila::Dnn::GqaConfig	inline
getHeadDim() const noexcept	Mila::Dnn::GqaConfig	inline
getModelDim() const noexcept	Mila::Dnn::GqaConfig	inline
getNumHeads() const noexcept	Mila::Dnn::GqaConfig	inline
getNumKvHeads() const noexcept	Mila::Dnn::GqaConfig	inline
GqaConfig(dim_t model_dim, dim_t num_heads, dim_t num_kv_heads)	Mila::Dnn::GqaConfig	inline
model_dim_	Mila::Dnn::GqaConfig	private
num_heads_	Mila::Dnn::GqaConfig	private
num_kv_heads_	Mila::Dnn::GqaConfig	private
toMetadata() const	Mila::Dnn::GqaConfig	inlinevirtual
toString() const override	Mila::Dnn::GqaConfig	inlinevirtual
validate() const override	Mila::Dnn::GqaConfig	inlinevirtual
withModelDim(this Self &&self, dim_t model_dim)	Mila::Dnn::GqaConfig	inline
withNumHeads(this Self &&self, dim_t num_heads)	Mila::Dnn::GqaConfig	inline
withNumKvHeads(this Self &&self, dim_t num_kv_heads)	Mila::Dnn::GqaConfig	inline
~ComponentConfig()=default	Mila::Dnn::ComponentConfig	virtual