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

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

fromMetadata(const SerializationMetadata &meta) overrideMila::Dnn::GptBlockConfiginlinevirtual
getActivationType() const noexcept (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
getEffectiveHiddenDimension() const noexceptMila::Dnn::GptBlockConfiginline
getHiddenSize() const noexcept (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
getModelDim() const noexcept (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
getNumHeads() const noexcept (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
getResidualScale() const noexcept (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
GptBlockConfig(dim_t model_dim, dim_t num_heads)Mila::Dnn::GptBlockConfiginline
toMetadata() const overrideMila::Dnn::GptBlockConfiginlinevirtual
toString() const overrideMila::Dnn::GptBlockConfiginlinevirtual
useBias() const noexcept (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
validate() const overrideMila::Dnn::GptBlockConfiginlinevirtual
withActivation(this Self &&self, ActivationType activation_type) (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
withBias(this Self &&self, bool use_bias) (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
withHiddenSize(this Self &&self, dim_t hidden_dim) (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
withMaxSequenceLength(this Self &&self, dim_t max_seq_len)Mila::Dnn::GptBlockConfiginline
withResidualScale(this Self &&self, float scale) (defined in Mila::Dnn::GptBlockConfig)Mila::Dnn::GptBlockConfiginline
~ComponentConfig()=defaultMila::Dnn::ComponentConfigvirtual