Mila
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) overrideMila::Dnn::MultiHeadAttentionConfiginlinevirtual
getModelDim() const noexceptMila::Dnn::MultiHeadAttentionConfiginline
getNumHeads() const noexceptMila::Dnn::MultiHeadAttentionConfiginline
MultiHeadAttentionConfig(dim_t model_dim, dim_t num_heads)Mila::Dnn::MultiHeadAttentionConfiginline
toMetadata() const overrideMila::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