Mila
0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Here is a list of all class members with links to the classes they belong to:
- t -
T_ :
Mila::Dnn::Compute::CpuAttentionOp
,
Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
TAccumPrecision :
Mila::Dnn::Compute::Cuda::CublasLtLinearPlan< TComputePrecision, TParameterPrecision >
target_data_type :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
TargetDataType :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
targets() :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
,
Mila::Data::TokenSequenceLoader< TMemoryResource >
TargetsHostType :
Mila::Dnn::Compute::CpuSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets >
TargetsNativeType :
Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets >
TargetsTensorType :
Mila::Dnn::Compute::CpuSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets >
,
Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets >
TargetTensor :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
TargetTensorType :
Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TLogits, TTargets, TPrecision >
temperature :
Mila::Dnn::GenerateParams
Tensor() :
Mila::Dnn::Tensor< TDataType, TMemoryResource >
tensor_index_ :
Mila::Dnn::Serialization::PretrainedModelReader
TensorBlob() :
Mila::Dnn::Serialization::TensorBlob< MR >
TensorBuffer() :
Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
TensorInputAType :
Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB >
TensorInputBType :
Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB >
TensorInputType :
Mila::Dnn::Compute::UnaryOperation< TDeviceType, TInput, TPrecision >
TensorLeftType :
Mila::Dnn::Compute::BinaryOperation< TDeviceType, TPrecision, TInputA, TInputB >
TensorOutputType :
Mila::Dnn::Compute::BinaryOperation< TDeviceType, TPrecision, TInputA, TInputB >
,
Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB >
,
Mila::Dnn::Compute::UnaryOperation< TDeviceType, TInput, TPrecision >
TensorRightType :
Mila::Dnn::Compute::BinaryOperation< TDeviceType, TPrecision, TInputA, TInputB >
TensorShape() :
Mila::Dnn::TensorShape
TensorType :
Mila::Data::TokenSequenceLoader< TMemoryResource >
,
Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
,
Mila::Dnn::Compute::CpuAttentionOp
,
Mila::Dnn::Compute::CpuEncoderOp
,
Mila::Dnn::Compute::CpuGeluOp
,
Mila::Dnn::Compute::CpuLayerNormOp
,
Mila::Dnn::Compute::CpuLinearOp
,
Mila::Dnn::Compute::CpuSoftmaxOp
,
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::Linear::CudaLinearOp< TComputePrecision, TWeightQuant >
,
Mila::Dnn::Compute::Cuda::Lpe::CudaLpeOp< TInput, TPrecision >
,
Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::Rope::CudaRopeOp< TComputePrecision >
,
Mila::Dnn::Compute::Cuda::Softmax::CudaSoftmaxOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::Swiglu::CudaSwigluOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::TokenEmbedding::CudaTokenEmbeddingOp< TInput, TPrecision >
,
Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
,
Mila::Dnn::Detail::mlp_activation_impl< ActivationType::Gelu, TDeviceType, TPrecision >
,
Mila::Dnn::Detail::mlp_activation_impl< ActivationType::Swiglu, TDeviceType, TPrecision >
,
Mila::Dnn::Gelu< TDeviceType, TPrecision >
,
Mila::Dnn::GptBlock< TDeviceType, TPrecision >
,
Mila::Dnn::GptModel< TDeviceType, TPrecision >
,
Mila::Dnn::GptTransformer< TDeviceType, TPrecision >
,
Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >
,
Mila::Dnn::LanguageNetwork< TDeviceType, TPrecision >
,
Mila::Dnn::LayerNorm< TDeviceType, TPrecision >
,
Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
,
Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
,
Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
,
Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
,
Mila::Dnn::Loss< TDeviceType, TPrecision >
,
Mila::Dnn::MLP< TDeviceType, TPrecision >
,
Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
,
Mila::Dnn::Residual< TDeviceType, TPrecision >
,
Mila::Dnn::RmsNorm< TDeviceType, TPrecision >
,
Mila::Dnn::Rope< TDeviceType, TPrecision >
,
Mila::Dnn::Softmax< TDeviceType, TPrecision >
,
Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TLogits, TTargets, TPrecision >
,
Mila::Dnn::Swiglu< TDeviceType, TPrecision >
to_float() :
Mila::Dnn::VulkanTensorTraits
toJson() :
Mila::Data::SerializationMetadata
,
Mila::Dnn::Optimizers::SerializationMetadata
,
Mila::Dnn::Serialization::SerializationMetadata
,
Mila::Dnn::SerializationMetadata
token_embed_out_ptr_ :
Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
token_embedding_ :
Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
token_to_id_ :
Mila::Data::BpeVocabulary
token_window_size :
Mila::Data::TokenSequenceLoaderConfig
TokenEmbedding() :
Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
TokenEmbeddingType :
Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
TokenIndexType :
Mila::Dnn::GptModel< TDeviceType, TPrecision >
,
Mila::Dnn::GptTransformer< TDeviceType, TPrecision >
,
Mila::Dnn::LanguageNetwork< TDeviceType, TPrecision >
,
Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
,
Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
,
Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >
,
Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
tokens_file_path_ :
Mila::Data::TokenSequenceLoader< TMemoryResource >
tokens_generated :
Mila::Dnn::GenerationStatistics
TokenSequenceLoader() :
Mila::Data::TokenSequenceLoader< TMemoryResource >
tokenToId() :
Mila::Data::BpeVocabulary
,
Mila::Data::CharVocabulary
,
Mila::Data::TokenizerVocabulary
tokenToString() :
Mila::Data::BpeTokenizer
,
Mila::Data::CharTokenizer
,
Mila::Data::Tokenizer
toMetadata() :
Mila::Data::BpeVocabularyConfig
,
Mila::Data::CharVocabularyConfig
,
Mila::Dnn::ComponentConfig
,
Mila::Dnn::CrossEntropyConfig
,
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
toString() :
Mila::Data::BpeVocabularyConfig
,
Mila::Data::CharVocabularyConfig
,
Mila::Dnn::Component< TDeviceType, TPrecision >
,
Mila::Dnn::ComponentConfig
,
Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
,
Mila::Dnn::Compute::DeviceId
,
Mila::Dnn::ConstantLRScheduler
,
Mila::Dnn::CosineLRScheduler
,
Mila::Dnn::CrossEntropyConfig
,
Mila::Dnn::Dropout< TDeviceType, TInput, TOutput >
,
Mila::Dnn::FusedComponent< TDeviceType, TPrecision >
,
Mila::Dnn::Gelu< TDeviceType, TPrecision >
,
Mila::Dnn::GeluConfig
,
Mila::Dnn::GptBlock< TDeviceType, TPrecision >
,
Mila::Dnn::GptBlockConfig
,
Mila::Dnn::GptConfig
,
Mila::Dnn::GptModel< TDeviceType, TPrecision >
,
Mila::Dnn::GptTransformer< TDeviceType, TPrecision >
,
Mila::Dnn::GqaConfig
,
Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >
,
Mila::Dnn::LayerNorm< TDeviceType, TPrecision >
,
Mila::Dnn::LayerNormConfig
,
Mila::Dnn::LearningRateScheduler
,
Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
,
Mila::Dnn::LinearConfig
,
Mila::Dnn::LinearLRScheduler
,
Mila::Dnn::LlamaConfig
,
Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
,
Mila::Dnn::LlamaModelConfig
,
Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
,
Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >
,
Mila::Dnn::LpeConfig
,
Mila::Dnn::MemoryStats
,
Mila::Dnn::MLP< TDeviceType, TPrecision >
,
Mila::Dnn::MLPConfig
,
Mila::Dnn::Model< TDeviceType, TPrecision >
,
Mila::Dnn::ModelConfig
,
Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
,
Mila::Dnn::MultiHeadAttentionConfig
,
Mila::Dnn::Network< TDeviceType, TPrecision >
,
Mila::Dnn::Optimizers::AdamWConfig
,
Mila::Dnn::Residual< TDeviceType, TPrecision >
,
Mila::Dnn::ResidualConfig
,
Mila::Dnn::RmsNorm< TDeviceType, TPrecision >
,
Mila::Dnn::RmsNormConfig
,
Mila::Dnn::Rope< TDeviceType, TPrecision >
,
Mila::Dnn::RopeConfig
,
Mila::Dnn::Softmax< TDeviceType, TPrecision >
,
Mila::Dnn::SoftmaxConfig
,
Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TLogits, TTargets, TPrecision >
,
Mila::Dnn::Swiglu< TDeviceType, TPrecision >
,
Mila::Dnn::SwigluConfig
,
Mila::Dnn::Tensor< TDataType, TMemoryResource >
,
Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
,
Mila::Dnn::TokenEmbeddingConfig
,
Mila::Version
total_bytes :
Mila::Dnn::Serialization::TensorMetadata
total_steps_ :
Mila::Dnn::CosineLRScheduler
,
Mila::Dnn::LinearLRScheduler
totalAllocated :
Mila::Dnn::Compute::MemoryStats
totalBytes() :
Mila::Dnn::MemoryStats
totalDeallocated :
Mila::Dnn::Compute::MemoryStats
totalDeviceBytes() :
Mila::Dnn::MemoryStats
totalGlobalMem :
Mila::Dnn::Compute::CudaDeviceProps
totalHostBytes() :
Mila::Dnn::MemoryStats
trace() :
Mila::Logging::Logger
TrackedMemoryResource() :
Mila::Dnn::Compute::TrackedMemoryResource
train() :
Mila::Data::BpeTrainer
,
Mila::Data::BpeVocabulary
,
Mila::Data::CharTrainer
,
Mila::Data::CharVocabulary
,
Mila::Data::TokenizerTrainer
,
Mila::Dnn::Model< TDeviceType, TPrecision >
trainFromFile() :
Mila::Data::BpeVocabulary
,
Mila::Data::CharVocabulary
training_mode_ :
Mila::Dnn::Component< TDeviceType, TPrecision >
,
Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
training_mode_mutex_ :
Mila::Dnn::Component< TDeviceType, TPrecision >
trainVocabulary() :
Mila::Data::TrainerFactory
transformer_blocks_ :
Mila::Dnn::GptTransformer< TDeviceType, TPrecision >
,
Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
TransformerBlockType :
Mila::Dnn::GptTransformer< TDeviceType, TPrecision >
,
Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
truncateIfNeeded() :
Mila::Dnn::GptModel< TDeviceType, TPrecision >
,
Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
tryFindComponent() :
Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
tryGetBool() :
Mila::Data::SerializationMetadata
,
Mila::Dnn::Optimizers::SerializationMetadata
,
Mila::Dnn::Serialization::SerializationMetadata
,
Mila::Dnn::SerializationMetadata
tryGetDouble() :
Mila::Data::SerializationMetadata
,
Mila::Dnn::Optimizers::SerializationMetadata
,
Mila::Dnn::Serialization::SerializationMetadata
,
Mila::Dnn::SerializationMetadata
tryGetFloat() :
Mila::Data::SerializationMetadata
,
Mila::Dnn::Optimizers::SerializationMetadata
,
Mila::Dnn::Serialization::SerializationMetadata
,
Mila::Dnn::SerializationMetadata
tryGetInt() :
Mila::Data::SerializationMetadata
,
Mila::Dnn::Optimizers::SerializationMetadata
,
Mila::Dnn::Serialization::SerializationMetadata
,
Mila::Dnn::SerializationMetadata
tryGetShape() :
Mila::Data::SerializationMetadata
,
Mila::Dnn::Optimizers::SerializationMetadata
,
Mila::Dnn::Serialization::SerializationMetadata
,
Mila::Dnn::SerializationMetadata
tryGetString() :
Mila::Data::SerializationMetadata
,
Mila::Dnn::Optimizers::SerializationMetadata
,
Mila::Dnn::Serialization::SerializationMetadata
,
Mila::Dnn::SerializationMetadata
type :
Mila::Dnn::Compute::DeviceId
,
Mila::Dnn::Compute::OperationTraits< OperationType::CrossEntropyOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::CrossEntropyOp, DeviceType::Cuda, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::GeluOp, DeviceType::Cpu, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::GeluOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::GeluOp, DeviceType::Cuda, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::GroupedQueryAttentionOp, DeviceType::Cuda, TensorDataType::BF16, NoKvCompression >
,
Mila::Dnn::Compute::OperationTraits< OperationType::GroupedQueryAttentionOp, DeviceType::Cuda, TensorDataType::FP32, NoKvCompression >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LinearOp, DeviceType::Cpu, TensorDataType::FP32, NoWeightQuant >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LinearOp, DeviceType::Cuda, TensorDataType::BF16, NoWeightQuant >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LinearOp, DeviceType::Cuda, TensorDataType::BF16, PerChannelFp8<> >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LinearOp, DeviceType::Cuda, TensorDataType::BF16, PerGroupFp4< 128 > >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LinearOp, DeviceType::Cuda, TensorDataType::BF16, PerGroupFp4< 64 > >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LinearOp, DeviceType::Cuda, TensorDataType::BF16, PerGroupInt4< 128 > >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LinearOp, DeviceType::Cuda, TensorDataType::BF16, PerGroupInt4< 64 > >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LinearOp, DeviceType::Cuda, TensorDataType::FP32, NoWeightQuant >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LpeOp, DeviceType::Cpu, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LpeOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::LpeOp, DeviceType::Cuda, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::MultiHeadAttentionOp, DeviceType::Cpu, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::MultiHeadAttentionOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::MultiHeadAttentionOp, DeviceType::Cuda, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::ResidualOp, DeviceType::Cpu, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::ResidualOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::ResidualOp, DeviceType::Cuda, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::RmsNormOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::RmsNormOp, DeviceType::Cuda, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::RopeOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::RopeOp, DeviceType::Cuda, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::SoftmaxOp, DeviceType::Cpu, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::SoftmaxOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::SoftmaxOp, DeviceType::Cuda, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::SwigluOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::SwigluOp, DeviceType::Cuda, TensorDataType::FP32, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::TokenEmbeddingOp, DeviceType::Cuda, TensorDataType::BF16, void >
,
Mila::Dnn::Compute::OperationTraits< OperationType::TokenEmbeddingOp, DeviceType::Cuda, TensorDataType::FP32, void >
type_name :
Mila::Dnn::TensorDataTypeMap< __nv_fp8_e4m3 >
,
Mila::Dnn::TensorDataTypeMap< __nv_fp8_e5m2 >
,
Mila::Dnn::TensorDataTypeMap< half >
,
Mila::Dnn::TensorDataTypeMap< nv_bfloat16 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::BF16 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP16 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP32 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP4_E2M1 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP4_E3M0 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP8_E4M3 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP8_E5M2 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::INT16 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::INT32 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::INT8 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::UINT16 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::UINT32 >
,
Mila::Dnn::TensorDataTypeTraits< TensorDataType::UINT8 >
Generated by
1.15.0