Here is a list of all functions with links to the classes they belong to:
- i -
- idToToken() : Mila::Data::BpeVocabulary, Mila::Data::CharVocabulary, Mila::Data::TokenizerVocabulary
- IExecutionContext() : Mila::Dnn::Compute::IExecutionContext
- incrementBatch() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
- indexToChar() : Mila::Data::CharVocabulary
- info() : Mila::Logging::Logger
- 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 >
- inputs() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >, Mila::Data::TokenSequenceLoader< TMemoryResource >
- inputShape() : Mila::Dnn::BuildContext
- 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_device_accessible() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
- is_device_only() : Mila::Dnn::VulkanTensorTraits
- is_host_accessible() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
- 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
- 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