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::RopeConfig
private
channels_
Mila::Dnn::RopeConfig
private
fromMetadata
(const SerializationMetadata &meta) override
Mila::Dnn::RopeConfig
inline
virtual
getBase
() const noexcept
Mila::Dnn::RopeConfig
inline
getEmbeddingDim
() const noexcept
Mila::Dnn::RopeConfig
inline
getHeadDim
() const noexcept
Mila::Dnn::RopeConfig
inline
getMaxSequenceLength
() const noexcept
Mila::Dnn::RopeConfig
inline
getNumHeads
() const noexcept
Mila::Dnn::RopeConfig
inline
getNumKVHeads
() const noexcept
Mila::Dnn::RopeConfig
inline
getRotaryDim
() const noexcept
Mila::Dnn::RopeConfig
inline
max_seq_len_
Mila::Dnn::RopeConfig
private
n_heads_
Mila::Dnn::RopeConfig
private
n_kv_heads_
Mila::Dnn::RopeConfig
private
RopeConfig
(size_t channels, size_t n_heads, size_t n_kv_heads, size_t max_seq_len)
Mila::Dnn::RopeConfig
inline
rotary_dim_
Mila::Dnn::RopeConfig
private
toMetadata
() const override
Mila::Dnn::RopeConfig
inline
virtual
toString
() const override
Mila::Dnn::RopeConfig
inline
virtual
validate
() const override
Mila::Dnn::RopeConfig
inline
virtual
withBase
(this Self &&self, float base)
Mila::Dnn::RopeConfig
inline
withRotaryDim
(this Self &&self, size_t rotary_dim)
Mila::Dnn::RopeConfig
inline
~ComponentConfig
()=default
Mila::Dnn::ComponentConfig
virtual
Generated by
1.15.0