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:
- i -
id_to_token_ :
Mila::Data::BpeVocabulary
idToToken() :
Mila::Data::BpeVocabulary
,
Mila::Data::CharVocabulary
,
Mila::Data::TokenizerVocabulary
idx_to_char_ :
Mila::Data::CharVocabulary
IExecutionContext() :
Mila::Dnn::Compute::IExecutionContext
impl_ :
Mila::Dnn::Compute::Cuda::Gelu::CudaGeluOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::Softmax::CudaSoftmaxOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::Swiglu::CudaSwigluOp< TPrecision >
,
Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
in_features_ :
Mila::Dnn::Compute::CpuLinearOp
includeSourceLocation_ :
Mila::Logging::ConsoleSink
,
Mila::Logging::FileSink
includeTimestamp_ :
Mila::Logging::ConsoleSink
,
Mila::Logging::FileSink
incrementBatch() :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
index :
Mila::Dnn::Compute::DeviceId
indexToChar() :
Mila::Data::CharVocabulary
info() :
Mila::Logging::Logger
initial_lr_ :
Mila::Dnn::CosineLRScheduler
,
Mila::Dnn::LinearLRScheduler
initialization_timeout_ms :
Mila::Data::TokenSequenceLoaderConfig
initialize_parameters_ :
Mila::Dnn::BuildContext
initializeBaseVocabulary() :
Mila::Data::BpeVocabulary
initializeDataset() :
Mila::Data::TokenSequenceLoader< TMemoryResource >
initializeExecutionResources() :
Mila::Dnn::Compute::ExecutionContext< DeviceType::Metal >
,
Mila::Dnn::Compute::ExecutionContext< DeviceType::Vulkan >
initializeGradients() :
Mila::Dnn::LayerNorm< TDeviceType, TPrecision >
,
Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
,
Mila::Dnn::RmsNorm< TDeviceType, TPrecision >
initializeKVCache() :
Mila::Dnn::GptBlock< TDeviceType, TPrecision >
initializeKvCache() :
Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
,
Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
,
Mila::Dnn::Compute::IKvCacheLifecycle
initializeParameterGradients() :
Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >
,
Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
initializeParameters() :
Mila::Dnn::LayerNorm< TDeviceType, TPrecision >
,
Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
,
Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >
,
Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
initializePreTokenization() :
Mila::Data::BpeTokenizer
initializeResources() :
Mila::Dnn::Compute::ExecutionContext< DeviceType::Cuda >
initializeState() :
Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
initializeState_optimized() :
Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
inner_size :
Mila::Dnn::AxisPartition
inner_size_ :
Mila::Dnn::Compute::CpuLayerNormOp
,
Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >
input_a_grad_ :
Mila::Dnn::Residual< TDeviceType, TPrecision >
input_b_grad_ :
Mila::Dnn::Residual< TDeviceType, TPrecision >
input_data_type :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
input_features_ :
Mila::Dnn::LinearConfig
,
Mila::Dnn::MLPConfig
input_grad_ :
Mila::Dnn::Gelu< TDeviceType, TPrecision >
,
Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >
,
Mila::Dnn::LayerNorm< TDeviceType, TPrecision >
,
Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
,
Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >
,
Mila::Dnn::RmsNorm< TDeviceType, TPrecision >
,
Mila::Dnn::Swiglu< TDeviceType, TPrecision >
,
Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
input_shape_ :
Mila::Dnn::BuildContext
,
Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
,
Mila::Dnn::Swiglu< TDeviceType, TPrecision >
InputDataType :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
inputs() :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
,
Mila::Data::TokenSequenceLoader< TMemoryResource >
inputShape() :
Mila::Dnn::BuildContext
InputTensor :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
instance() :
Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry
,
Mila::Dnn::Compute::DeviceRegistrar
,
Mila::Dnn::Compute::DeviceRegistry
,
Mila::Dnn::Compute::OperationRegistry
,
Mila::Dnn::Compute::OperationsRegistrar
,
Mila::Dnn::Extensibility::PluginManager
is_built_ :
Mila::Dnn::Compute::CpuAttentionOp
,
Mila::Dnn::Compute::CpuEncoderOp
,
Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
is_cache_coherent :
Mila::Dnn::Compute::MemoryResourceTraits< CpuMemoryResource >
is_cpu_resource :
Mila::Dnn::Compute::MemoryResourceTraits< TMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CpuMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaDeviceMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaManagedMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
is_cuda_resource :
Mila::Dnn::Compute::MemoryResourceTraits< TMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CpuMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaDeviceMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaManagedMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
is_device_accessible :
Mila::Dnn::Compute::CpuMemoryResource
,
Mila::Dnn::Compute::CudaDeviceMemoryResource
,
Mila::Dnn::Compute::CudaManagedMemoryResource
,
Mila::Dnn::Compute::CudaPinnedMemoryResource
,
Mila::Dnn::Compute::DeviceAccessible
,
Mila::Dnn::Compute::MemoryResource
,
Mila::Dnn::Compute::MetalMemoryResource
,
Mila::Dnn::Compute::VulkanMemoryResource
,
Mila::Dnn::Tensor< TDataType, TMemoryResource >
is_device_only :
Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
,
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 >
,
Mila::Dnn::VulkanTensorTraits
is_float_type :
Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
,
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 >
is_host_accessible :
Mila::Dnn::Compute::CpuMemoryResource
,
Mila::Dnn::Compute::CudaDeviceMemoryResource
,
Mila::Dnn::Compute::CudaManagedMemoryResource
,
Mila::Dnn::Compute::CudaPinnedMemoryResource
,
Mila::Dnn::Compute::HostAccessible
,
Mila::Dnn::Compute::MemoryResource
,
Mila::Dnn::Compute::MetalMemoryResource
,
Mila::Dnn::Compute::VulkanMemoryResource
,
Mila::Dnn::Tensor< TDataType, TMemoryResource >
is_initialized_ :
Mila::Dnn::Compute::OperationsRegistrar
is_integer_type :
Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
,
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 >
is_metal_resource :
Mila::Dnn::Compute::MemoryResourceTraits< TMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CpuMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaDeviceMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaManagedMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
is_mixed_precision :
Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
is_new :
Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry::AcquireResult
is_opencl_resource :
Mila::Dnn::Compute::MemoryResourceTraits< TMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CpuMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaDeviceMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaManagedMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
is_page_locked :
Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
is_training_ :
Mila::Data::TokenSequenceLoader< TMemoryResource >
is_view_ :
Mila::Dnn::Tensor< TDataType, TMemoryResource >
is_vulkan_resource :
Mila::Dnn::Compute::MemoryResourceTraits< TMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CpuMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaDeviceMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaManagedMemoryResource >
,
Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
isAbsolutePath() :
Mila::Dnn::Serialization::ModelArchive
isAligned() :
Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
isAvailable() :
Mila::Dnn::Compute::MetalDevicePlugin
isBf16Supported() :
Mila::Dnn::Compute::CudaDevice
isBuilt() :
Mila::Dnn::Component< TDeviceType, TPrecision >
,
Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
isByteLevel() :
Mila::Data::BpeVocabulary
,
Mila::Data::BpeVocabularyConfig
,
Mila::Data::CharVocabularyConfig
isCaseSensitive() :
Mila::Data::CharVocabularyConfig
isClosed() :
Mila::Dnn::Serialization::ModelArchive
isCompatible() :
Mila::Dnn::Extensibility::PluginManager
isCpuDevice() :
Mila::Dnn::Compute::ExecutionContext< DeviceType::Metal >
,
Mila::Dnn::Compute::ExecutionContext< DeviceType::Vulkan >
isDeviceUsable() :
Mila::Dnn::Compute::CudaDeviceRegistrar
,
Mila::Dnn::Compute::MetalDevicePlugin
isDiscreteGpu() :
Mila::Dnn::Compute::VulkanDevice
isEnabled() :
Mila::Data::SpecialTokens
,
Mila::Logging::ConsoleSink
,
Mila::Logging::FileSink
,
Mila::Logging::Logger
,
Mila::Logging::NullSink
isEval() :
Mila::Dnn::Model< TDeviceType, TPrecision >
isEvalMode() :
Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
isFp16Supported() :
Mila::Dnn::Compute::CudaDevice
isFp8Supported() :
Mila::Dnn::Compute::CudaDevice
isIdentifier() :
Mila::Dnn::Component< TDeviceType, TPrecision >
isInferenceMode() :
Mila::Dnn::BuildContext
,
Mila::Dnn::Component< TDeviceType, TPrecision >
,
Mila::Dnn::Model< TDeviceType, TPrecision >
isInt8Supported() :
Mila::Dnn::Compute::CudaDevice
isMergeCachingEnabled() :
Mila::Data::BpeVocabularyConfig
isMetalDevice() :
Mila::Dnn::Compute::ExecutionContext< DeviceType::Metal >
isMPSAvailable() :
Mila::Dnn::Compute::MetalDevicePlugin
isMPSSupported() :
Mila::Dnn::Compute::MetalDevice
isOpen() :
Mila::Dnn::Serialization::PretrainedModelReader
,
Mila::Dnn::Serialization::Serializer
,
Mila::Dnn::Serialization::ZipSerializer
isOpenForRead() :
Mila::Dnn::Serialization::Serializer
,
Mila::Dnn::Serialization::ZipSerializer
isOpenForWrite() :
Mila::Dnn::Serialization::Serializer
,
Mila::Dnn::Serialization::ZipSerializer
isOperationRegistered() :
Mila::Dnn::Compute::OperationRegistry
isRegistered :
Mila::Dnn::Compute::CpuAttentionOpRegistrar
,
Mila::Dnn::Compute::CpuCrossEntropyOpRegistrar
,
Mila::Dnn::Compute::CpuLinearOpRegistrar
,
Mila::Dnn::Compute::CpuSoftmaxCrossEntropyOpRegistrar
,
Mila::Dnn::Compute::CpuSoftmaxOpRegistrar
,
Mila::Dnn::Compute::Cuda::Linear::CudaLinearOpRegistrar
,
Mila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOpRegistrar
isScalar() :
Mila::Dnn::ITensor
,
Mila::Dnn::Tensor< TDataType, TMemoryResource >
isSpecialToken() :
Mila::Data::SpecialTokens
isTrainingMode() :
Mila::Dnn::BuildContext
,
Mila::Dnn::Component< TDeviceType, TPrecision >
,
Mila::Dnn::Model< TDeviceType, TPrecision >
isType() :
Mila::Dnn::ITensor
isUnifiedMemorySupported() :
Mila::Dnn::Compute::MetalDevice
isValid() :
Mila::Dnn::Compute::Cuda::CublasLtLinearPlan< TComputePrecision, TParameterPrecision >
,
Mila::Dnn::Compute::Cuda::CublasLtMatMulPlan< TComputePrecision >
,
Mila::Dnn::Tensor< TDataType, TMemoryResource >
isValidToken() :
Mila::Data::BpeTokenizer
,
Mila::Data::CharTokenizer
,
Mila::Data::Tokenizer
isView() :
Mila::Dnn::ITensor
,
Mila::Dnn::Tensor< TDataType, TMemoryResource >
isVulkanDevice() :
Mila::Dnn::Compute::ExecutionContext< DeviceType::Vulkan >
item() :
Mila::Dnn::Tensor< TDataType, TMemoryResource >
ITensorBlob() :
Mila::Dnn::Serialization::ITensorBlob
Generated by
1.15.0