Loading...
Searching...
No Matches
Here is a list of all variables with links to the classes they belong to:
- n -
- n_heads : Mila::Dnn::Visualization::VisualizerContext
- n_heads_ : Mila::Dnn::RopeConfig
- n_kv_heads_ : Mila::Dnn::RopeConfig
- name : Mila::Dnn::Compute::CudaDeviceProps, Mila::Dnn::Extensibility::PluginInfo, Mila::Dnn::Serialization::TensorBlobMetadata
- name_ : Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::Compute::TrackedMemoryResource, Mila::Dnn::Tensor< TDataType, TMemoryResource >
- nbytes : Mila::Dnn::Serialization::TensorBlobMetadata
- ndim : Mila::Dnn::TensorShape
- network_ : Mila::Dnn::Model< TDeviceType, TPrecision >
- NH_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
- NKV_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
- norm_axis_ : Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >, Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >
- norm_dim_ : Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >
- norm_epsilon : Mila::Dnn::Serialization::PretrainedMetadata
- norm_type : Mila::Dnn::Serialization::PretrainedMetadata
- normalize_unicode_ : Mila::Data::CharVocabularyConfig
- normalized_axis : Mila::Dnn::AxisPartition
- normalized_ptr_ : Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
- normalized_shape : Mila::Dnn::MultiAxisPartition
- normalized_shape_ : Mila::Dnn::LayerNormConfig, Mila::Dnn::RmsNormConfig
- normalized_size : Mila::Dnn::MultiAxisPartition
- num_batches_ : Mila::Data::TokenSequenceLoader< TMemoryResource >
- num_heads : Mila::Dnn::Serialization::PretrainedMetadata
- num_heads_ : Mila::Dnn::GptBlockConfig, Mila::Dnn::GptConfig, Mila::Dnn::GqaConfig, Mila::Dnn::LlamaConfig, Mila::Dnn::MultiHeadAttentionConfig
- num_kv_heads : Mila::Dnn::Serialization::PretrainedMetadata
- num_kv_heads_ : Mila::Dnn::GqaConfig, Mila::Dnn::LlamaConfig
- num_layers : Mila::Dnn::Serialization::PretrainedMetadata
- num_layers_ : Mila::Dnn::GptConfig, Mila::Dnn::LlamaConfig
- num_slices : Mila::Dnn::AxisPartition
- num_tensors_ : Mila::Dnn::Serialization::PretrainedModelReader
- num_tokens_ : Mila::Data::TokenSequenceLoader< TMemoryResource >
- num_windows_ : Mila::Data::TokenSequenceLoader< TMemoryResource >