Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
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::GqaConfiginlinevirtual
getGroupSize() const noexceptMila::Dnn::GqaConfiginline
getHeadDim() const noexceptMila::Dnn::GqaConfiginline
getModelDim() const noexceptMila::Dnn::GqaConfiginline
getNumHeads() const noexceptMila::Dnn::GqaConfiginline
getNumKvHeads() const noexceptMila::Dnn::GqaConfiginline
GqaConfig(dim_t model_dim, dim_t num_heads, dim_t num_kv_heads)Mila::Dnn::GqaConfiginline
model_dim_Mila::Dnn::GqaConfigprivate
num_heads_Mila::Dnn::GqaConfigprivate
num_kv_heads_Mila::Dnn::GqaConfigprivate
toMetadata() constMila::Dnn::GqaConfiginlinevirtual
toString() const overrideMila::Dnn::GqaConfiginlinevirtual
validate() const overrideMila::Dnn::GqaConfiginlinevirtual
withModelDim(this Self &&self, dim_t model_dim)Mila::Dnn::GqaConfiginline
withNumHeads(this Self &&self, dim_t num_heads)Mila::Dnn::GqaConfiginline
withNumKvHeads(this Self &&self, dim_t num_kv_heads)Mila::Dnn::GqaConfiginline
~ComponentConfig()=defaultMila::Dnn::ComponentConfigvirtual