Here is a list of all class members 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::Compute::TrackedMemoryResource, Mila::Dnn::Extensibility::PluginInfo, Mila::Dnn::Serialization::TensorBlobMetadata
- name_ : Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::Compute::TrackedMemoryResource, Mila::Dnn::Tensor< TDataType, TMemoryResource >
- native_type : Mila::Dnn::CpuTensorDataTypeTraits, Mila::Dnn::VulkanTensorTraits
- NativeType : Mila::Dnn::Compute::Cuda::Gelu::CudaGeluOp< TPrecision >, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >, Mila::Dnn::Compute::Cuda::Lpe::CudaLpeOp< TInput, TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >, Mila::Dnn::Compute::Cuda::Residual::CudaResidualOp< TInputA, TInputB, TPrecision >, Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >, Mila::Dnn::Compute::Cuda::Softmax::CudaSoftmaxOp< TPrecision >, Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets >, Mila::Dnn::Compute::Cuda::Swiglu::CudaSwigluOp< TPrecision >, Mila::Dnn::Compute::Cuda::TokenEmbedding::CudaTokenEmbeddingOp< TInput, TPrecision >, Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
- nbytes : Mila::Dnn::Serialization::TensorBlobMetadata
- ndim : Mila::Dnn::TensorShape
- Network : Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::Network< TDeviceType, TPrecision >
- network_ : Mila::Dnn::Model< TDeviceType, TPrecision >
- NetworkBase : Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::LanguageNetwork< TDeviceType, TPrecision >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
- NetworkFactoryFunc : Mila::Dnn::NetworkFactory
- NetworkType : Mila::Dnn::Model< TDeviceType, TPrecision >
- nextBatch() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >, Mila::Data::TokenSequenceLoader< TMemoryResource >
- 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 >
- none() : Mila::Data::SpecialTokens
- 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
- normalizeText() : Mila::Data::CharVocabulary
- normalizeZipPath() : Mila::Dnn::Serialization::ZipSerializer
- NullSink() : Mila::Logging::NullSink
- 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 >
- numBatches() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >, Mila::Data::TokenSequenceLoader< TMemoryResource >
- numTokens() : Mila::Data::TokenSequenceLoader< TMemoryResource >
- numWindows() : Mila::Data::TokenSequenceLoader< TMemoryResource >
- NvtxRange() : Mila::Profiling::NvtxRange