Mila 0.13.48
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.

base_Mila::Dnn::RopeConfigprivate
channels_Mila::Dnn::RopeConfigprivate
fromMetadata(const SerializationMetadata &meta) overrideMila::Dnn::RopeConfiginlinevirtual
getBase() const noexceptMila::Dnn::RopeConfiginline
getEmbeddingDim() const noexceptMila::Dnn::RopeConfiginline
getHeadDim() const noexceptMila::Dnn::RopeConfiginline
getMaxSequenceLength() const noexceptMila::Dnn::RopeConfiginline
getNumHeads() const noexceptMila::Dnn::RopeConfiginline
getNumKVHeads() const noexceptMila::Dnn::RopeConfiginline
getRotaryDim() const noexceptMila::Dnn::RopeConfiginline
max_seq_len_Mila::Dnn::RopeConfigprivate
n_heads_Mila::Dnn::RopeConfigprivate
n_kv_heads_Mila::Dnn::RopeConfigprivate
RopeConfig(size_t channels, size_t n_heads, size_t n_kv_heads, size_t max_seq_len)Mila::Dnn::RopeConfiginline
rotary_dim_Mila::Dnn::RopeConfigprivate
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, size_t rotary_dim)Mila::Dnn::RopeConfiginline
~ComponentConfig()=defaultMila::Dnn::ComponentConfigvirtual