Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::MultiHeadAttentionConfig Member List

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

fromMetadata(const SerializationMetadata &meta)Mila::Dnn::MultiHeadAttentionConfiginlinevirtual
getModelDim() const noexceptMila::Dnn::MultiHeadAttentionConfiginline
getNumHeads() const noexceptMila::Dnn::MultiHeadAttentionConfiginline
model_dim_Mila::Dnn::MultiHeadAttentionConfigprivate
MultiHeadAttentionConfig(dim_t model_dim, dim_t num_heads)Mila::Dnn::MultiHeadAttentionConfiginline
num_heads_Mila::Dnn::MultiHeadAttentionConfigprivate
toMetadata() constMila::Dnn::MultiHeadAttentionConfiginlinevirtual
toString() const overrideMila::Dnn::MultiHeadAttentionConfiginlinevirtual
validate() const overrideMila::Dnn::MultiHeadAttentionConfiginlinevirtual
withModelDim(this Self &&self, dim_t model_dim)Mila::Dnn::MultiHeadAttentionConfiginline
withNumHeads(this Self &&self, dim_t num_heads)Mila::Dnn::MultiHeadAttentionConfiginline
~ComponentConfig()=defaultMila::Dnn::ComponentConfigvirtual