Here is a list of all class members with links to the classes they belong to:

- d -

• d_input_ : Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• d_res1_accum_ : Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• data() : Mila::Dnn::Serialization::ITensorBlob, Mila::Dnn::Serialization::TensorBlob< MR >, Mila::Dnn::Tensor< TDataType, TMemoryResource >, Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >, Mila::Dnn::TensorShape
• data_ : Mila::Data::SerializationMetadata, Mila::Dnn::Optimizers::SerializationMetadata, Mila::Dnn::Serialization::SerializationMetadata, Mila::Dnn::SerializationMetadata, Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• data_buffer : Mila::Dnn::Serialization::TensorBlob< MR >
• data_type : Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >, Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >, Mila::Dnn::TensorDataTypeMap< __nv_fp8_e4m3 >, Mila::Dnn::TensorDataTypeMap< __nv_fp8_e5m2 >, Mila::Dnn::TensorDataTypeMap< float >, Mila::Dnn::TensorDataTypeMap< half >, Mila::Dnn::TensorDataTypeMap< nv_bfloat16 >, Mila::Dnn::TensorDataTypeMap< std::int16_t >, Mila::Dnn::TensorDataTypeMap< std::int32_t >, Mila::Dnn::TensorDataTypeMap< std::int8_t >, Mila::Dnn::TensorDataTypeMap< std::uint16_t >, Mila::Dnn::TensorDataTypeMap< std::uint32_t >, Mila::Dnn::TensorDataTypeMap< std::uint8_t >
• data_type_a : Mila::Dnn::Compute::OperationRegistry::TypeID
• data_type_b : Mila::Dnn::Compute::OperationRegistry::TypeID
• DataLoader() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
• DataType : Mila::Dnn::Tensor< TDataType, TMemoryResource >, Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• DataTypeTraits : Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >, Mila::Dnn::Tensor< TDataType, TMemoryResource >, Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• datt_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• datt_tensor_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• deallocationCount : Mila::Dnn::Compute::MemoryStats
• debug() : Mila::Logging::Logger
• decode() : Mila::Data::BpeTokenizer, Mila::Data::CharTokenizer, Mila::Data::Tokenizer, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::Linear::Detail::cuda_matvec_impl< float, float >, Mila::Dnn::Compute::Cuda::Linear::Detail::cuda_matvec_impl< nv_bfloat16, __nv_fp8_e4m3 >, Mila::Dnn::Compute::Cuda::Linear::Detail::cuda_matvec_impl< nv_bfloat16, nv_bfloat16 >, Mila::Dnn::Compute::Cuda::Lpe::CudaLpeOp< TInput, TPrecision >, Mila::Dnn::Compute::Cuda::Lpe::Detail::cuda_lpe_impl< float >, Mila::Dnn::Compute::Cuda::Lpe::Detail::cuda_lpe_impl< half >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >, Mila::Dnn::Compute::Cuda::Rope::CudaRopeOp< TComputePrecision >, Mila::Dnn::Compute::Cuda::Rope::Detail::cuda_rope_impl< __nv_bfloat16 >, Mila::Dnn::Compute::Cuda::Rope::Detail::cuda_rope_impl< float >, Mila::Dnn::Compute::Cuda::TokenEmbedding::CudaTokenEmbeddingOp< TInput, TPrecision >, Mila::Dnn::Compute::Cuda::TokenEmbedding::Detail::cuda_token_embedding_impl< __nv_bfloat16 >, Mila::Dnn::Compute::Cuda::TokenEmbedding::Detail::cuda_token_embedding_impl< float >, Mila::Dnn::Compute::IKvInference, Mila::Dnn::Compute::IPackedKvInference, Mila::Dnn::Compute::IPositionalDecode, Mila::Dnn::Compute::IPositionalPairedOp, Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::LanguageNetwork< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >, Mila::Dnn::MLP< TDeviceType, TPrecision >, Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >, Mila::Dnn::Rope< TDeviceType, TPrecision >
• decode_active_ : Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
• decode_optimized() : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• decode_output_ : Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >
• decode_path_ : Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >
• decode_time_ms : Mila::Dnn::GenerationStatistics
• decode_token_device_ : Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
• decode_token_staging_ : Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
• decode_tokens_per_second : Mila::Dnn::GenerationStatistics
• decodeImpl() : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• decodeToken() : Mila::Data::BpeTokenizer
• defaultLogger() : Mila::Logging::Logger
• defaultLogger_ : Mila::Logging::Logger
• device_ : Mila::Data::TokenSequenceLoader< TMemoryResource >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Metal >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Vulkan >
• device_accessible : Mila::Dnn::Compute::MemoryResourceTraits< CudaDeviceMemoryResource >, Mila::Dnn::Compute::MemoryResourceTraits< CudaManagedMemoryResource >, Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
• device_cache_ : Mila::Dnn::Compute::DeviceRegistry
• device_gradient_bytes : Mila::Dnn::MemoryStats
• device_id : Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry::CacheKey
• device_id_ : Mila::Dnn::Compute::CudaDevice, Mila::Dnn::Compute::CudaDeviceMemoryResource, Mila::Dnn::Compute::CudaDeviceProps, Mila::Dnn::Compute::CudaManagedMemoryResource, Mila::Dnn::Compute::CudaPinnedMemoryResource, Mila::Dnn::Compute::ExecutionContext< DeviceType::Cpu >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Cuda >, Mila::Dnn::Tensor< TDataType, TMemoryResource >, Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• device_index_ : Mila::Dnn::Compute::MetalDevice, Mila::Dnn::Compute::VulkanDevice
• device_name_ : Mila::Dnn::Compute::MetalDevice, Mila::Dnn::Compute::VulkanDevice
• device_parameter_bytes : Mila::Dnn::MemoryStats
• device_scratch_buf_ : Mila::Dnn::Compute::ExecutionContext< DeviceType::Cuda >
• device_scratch_size_ : Mila::Dnn::Compute::ExecutionContext< DeviceType::Cuda >
• device_state_bytes : Mila::Dnn::MemoryStats
• device_type : Mila::Dnn::Compute::CpuMemoryResource, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::BF16 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::FP16 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::FP32 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::FP4_E2M1 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::FP4_E3M0 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::FP8_E4M3 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::FP8_E5M2 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::INT16 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::INT32 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::INT8 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::UINT16 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::UINT32 >, Mila::Dnn::Compute::Cuda::TensorDataTypeMap< TensorDataType::UINT8 >, Mila::Dnn::Compute::CudaDeviceMemoryResource, Mila::Dnn::Compute::CudaManagedMemoryResource, Mila::Dnn::Compute::CudaPinnedMemoryResource, Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >, Mila::Dnn::Compute::OperationRegistry::TypeID
• device_type_name() : Mila::Dnn::VulkanTensorTraits
• DeviceConstructionKey() : Mila::Dnn::Compute::DeviceConstructionKey
• DeviceFactory : Mila::Dnn::Compute::DeviceRegistry
• DeviceId() : Mila::Dnn::Compute::DeviceId
• DevicePreference : Mila::Dnn::Compute::VulkanMemoryResource
• DeviceRegistrar() : Mila::Dnn::Compute::DeviceRegistrar
• DeviceRegistry : Mila::Dnn::Compute::DeviceConstructionKey, Mila::Dnn::Compute::DeviceRegistry
• dim_size_ : Mila::Dnn::Compute::CpuLayerNormOp
• dims : Mila::Dnn::TensorShape
• dispatchForward() : Mila::Dnn::Compute::Cuda::Rope::CudaRopeOp< TComputePrecision >
• diverging : Mila::Dnn::Visualization::ColorLUT
• divide() : Mila::Dnn::Compute::Cpu::MathOps, Mila::Dnn::Compute::Cuda::MathOps
• divideImpl() : Mila::Dnn::Compute::Cuda::MathOps
• dK_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• dk_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• dK_exp_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• dK_exp_tensor_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• dK_tensor_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• dk_tensor_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• do_allocate() : Mila::Dnn::Compute::CpuMemoryResource, Mila::Dnn::Compute::CudaDeviceMemoryResource, Mila::Dnn::Compute::CudaManagedMemoryResource, Mila::Dnn::Compute::CudaPinnedMemoryResource, Mila::Dnn::Compute::MemoryResource, Mila::Dnn::Compute::MetalMemoryResource, Mila::Dnn::Compute::TrackedMemoryResource, Mila::Dnn::Compute::VulkanMemoryResource
• do_deallocate() : Mila::Dnn::Compute::CpuMemoryResource, Mila::Dnn::Compute::CudaDeviceMemoryResource, Mila::Dnn::Compute::CudaManagedMemoryResource, Mila::Dnn::Compute::CudaPinnedMemoryResource, Mila::Dnn::Compute::MemoryResource, Mila::Dnn::Compute::MetalMemoryResource, Mila::Dnn::Compute::TrackedMemoryResource, Mila::Dnn::Compute::VulkanMemoryResource
• do_is_equal() : Mila::Dnn::Compute::CpuMemoryResource, Mila::Dnn::Compute::CudaDeviceMemoryResource, Mila::Dnn::Compute::CudaManagedMemoryResource, Mila::Dnn::Compute::CudaPinnedMemoryResource, Mila::Dnn::Compute::MemoryResource, Mila::Dnn::Compute::MetalMemoryResource, Mila::Dnn::Compute::TrackedMemoryResource, Mila::Dnn::Compute::VulkanMemoryResource
• dpreatt_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• dpreatt_tensor_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• dq_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• dq_tensor_ : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• Dropout() : Mila::Dnn::Dropout< TDeviceType, TInput, TOutput >
• DropoutConfig() : Mila::Dnn::DropoutConfig
• dtype : Mila::Dnn::Serialization::TensorBlobMetadata, Mila::Dnn::Serialization::TensorMetadata
• dummy_target_grad_ : Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TLogits, TTargets, TPrecision >
• dV_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• dv_ : Mila::Dnn::Compute::CpuAttentionOp
• dV_exp_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• dV_exp_tensor_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• dV_tensor_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• dv_tensor_ : Mila::Dnn::Compute::CpuAttentionOp
• dVout_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• dvout_ : Mila::Dnn::Compute::CpuAttentionOp
• dVout_tensor_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• dvout_tensor_ : Mila::Dnn::Compute::CpuAttentionOp