Here is a list of all class members with links to the classes they belong to:
- v -
- v_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
- v_data_ : Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >, Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
- v_exp_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
- v_exp_tensor_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
- v_opt_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
- v_out : Mila::Dnn::Compute::GqaState
- v_out_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
- v_out_decode : Mila::Dnn::Compute::GqaState
- v_out_decode_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
- v_out_decode_opt_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
- v_out_decode_tensor_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
- v_out_opt_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
- v_out_tensor_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
- v_out_tensor_optimized_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
- v_proj : Mila::Dnn::Visualization::VisualizerContext
- v_states_ : Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >, Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
- v_tensor_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
- valid() : Mila::Dnn::GenerationStatistics
- validate() : Mila::Data::BpeVocabularyConfig, Mila::Data::CharVocabularyConfig, Mila::Dnn::ComponentConfig, Mila::Dnn::CrossEntropyConfig, Mila::Dnn::DropoutConfig, Mila::Dnn::GeluConfig, Mila::Dnn::GptBlockConfig, Mila::Dnn::GptConfig, Mila::Dnn::GqaConfig, Mila::Dnn::LayerNormConfig, Mila::Dnn::LinearConfig, Mila::Dnn::LlamaConfig, Mila::Dnn::LpeConfig, Mila::Dnn::MLPConfig, Mila::Dnn::MultiHeadAttentionConfig, Mila::Dnn::Optimizers::AdamWConfig, Mila::Dnn::ResidualConfig, Mila::Dnn::RmsNormConfig, Mila::Dnn::RopeConfig, Mila::Dnn::SoftmaxConfig, Mila::Dnn::SwigluConfig, Mila::Dnn::TokenEmbeddingConfig
- validateBuildContext() : Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::LayerNorm< TDeviceType, TPrecision >, Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::Rope< TDeviceType, TPrecision >, Mila::Dnn::Swiglu< TDeviceType, TPrecision >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
- validateConcatenatedQKVShape() : Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
- validateCurrentBatch() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
- validateDecodeInputShape() : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
- validateDevice() : Mila::Dnn::Compute::ExecutionContext< DeviceType::Metal >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Vulkan >
- validateDeviceId() : Mila::Data::TokenSequenceLoader< TMemoryResource >, Mila::Dnn::Compute::CudaDevice, Mila::Dnn::Compute::ExecutionContext< DeviceType::Cuda >, Mila::Dnn::Tensor< TDataType, TMemoryResource >
- validateDeviceIndex() : Mila::Dnn::Compute::VulkanDevice
- validateHyperparameters() : Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
- validateIndices() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
- validateInputShape() : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::CpuEncoderOp, Mila::Dnn::Compute::CpuLayerNormOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::Lpe::CudaLpeOp< TInput, TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >, Mila::Dnn::Compute::Cuda::TokenEmbedding::CudaTokenEmbeddingOp< TInput, TPrecision >, Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::LayerNorm< TDeviceType, TPrecision >, Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >, Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >, Mila::Dnn::MLP< TDeviceType, TPrecision >, Mila::Dnn::RmsNorm< TDeviceType, TPrecision >, Mila::Dnn::Softmax< TDeviceType, TPrecision >, Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TLogits, TTargets, TPrecision >
- validateLeadingShape() : Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
- validateMetadata_() : Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::Swiglu< TDeviceType, TPrecision >
- validateName() : Mila::Dnn::Component< TDeviceType, TPrecision >
- validateNormalizedShape_() : Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >
- validatePrefillInputShape() : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
- validateRuntimeShape() : Mila::Dnn::Compute::Cuda::Rope::CudaRopeOp< TComputePrecision >, Mila::Dnn::Compute::Cuda::TokenEmbedding::CudaTokenEmbeddingOp< TInput, TPrecision >
- validateRuntimeShape_() : Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >
- validateShapeCompatibility() : Mila::Dnn::Compute::Cpu::MathOps
- validateShapeConsistency() : Mila::Dnn::Compute::CpuLayerNormOp
- value : Mila::Dnn::Compute::CudaDataTypeMap< __nv_bfloat16 >, Mila::Dnn::Compute::CudaDataTypeMap< float >, Mila::Dnn::Compute::CudaDataTypeMap< half >
- verbose_logging : Mila::Data::TokenSequenceLoaderConfig
- verifyArchitectureCompatibility() : Mila::Dnn::Network< TDeviceType, TPrecision >
- VERSION : Mila::Data::MilaFileHeader, Mila::Dnn::Serialization::PretrainedModelReader
- Version() : Mila::Version
- version : Mila::Dnn::Extensibility::PluginInfo
- view() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
- view_offset_ : Mila::Dnn::Tensor< TDataType, TMemoryResource >
- vocab_ : Mila::Data::BpeTokenizer, Mila::Data::CharTokenizer
- vocab_len_ : Mila::Dnn::LpeConfig
- vocab_size : Mila::Dnn::Serialization::PretrainedMetadata
- vocab_size_ : Mila::Data::BpeVocabularyConfig, Mila::Dnn::Compute::Cuda::TokenEmbedding::CudaTokenEmbeddingOp< TInput, TPrecision >, Mila::Dnn::CrossEntropyConfig, Mila::Dnn::GptConfig, Mila::Dnn::LlamaConfig, Mila::Dnn::TokenEmbeddingConfig
- vocabSize() : Mila::Dnn::GptModel< TDeviceType, TPrecision >, Mila::Dnn::LanguageModel< TDeviceType, TPrecision >, Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
- VulkanDevice() : Mila::Dnn::Compute::VulkanDevice
- VulkanMemoryResource() : Mila::Dnn::Compute::VulkanMemoryResource