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

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

fromMetadata(const SerializationMetadata &meta) overrideMila::Dnn::RopeConfiginlinevirtual
getBase() const noexcept (defined in Mila::Dnn::RopeConfig)Mila::Dnn::RopeConfiginline
getEmbeddingDim() const noexcept (defined in Mila::Dnn::RopeConfig)Mila::Dnn::RopeConfiginline
getHeadDim() const noexceptMila::Dnn::RopeConfiginline
getMaxSequenceLength() const noexceptMila::Dnn::RopeConfiginline
getNumHeads() const noexcept (defined in Mila::Dnn::RopeConfig)Mila::Dnn::RopeConfiginline
getNumKVHeads() const noexcept (defined in Mila::Dnn::RopeConfig)Mila::Dnn::RopeConfiginline
getRotaryDim() const noexcept (defined in Mila::Dnn::RopeConfig)Mila::Dnn::RopeConfiginline
RopeConfig(dim_t channels, dim_t n_heads, dim_t n_kv_heads, dim_t max_seq_len)Mila::Dnn::RopeConfiginline
toMetadata() const overrideMila::Dnn::RopeConfiginlinevirtual
toString() const overrideMila::Dnn::RopeConfiginlinevirtual
validate() const overrideMila::Dnn::RopeConfiginlinevirtual
withBase(this Self &&self, float base)Mila::Dnn::RopeConfiginline
withRotaryDim(this Self &&self, dim_t rotary_dim)Mila::Dnn::RopeConfiginline
~ComponentConfig()=defaultMila::Dnn::ComponentConfigvirtual