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

- o -

• onBuilding() : Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::FusedComponent< TDeviceType, TPrecision >, Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, 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::Lpe< TDeviceType, TIndex, 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 >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
• onExecutionContextSet() : Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >, 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::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::Swiglu< TDeviceType, TPrecision >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
• onGenerating() : Mila::Dnn::GptModel< TDeviceType, TPrecision >, Mila::Dnn::LanguageModel< TDeviceType, TPrecision >, Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
• onTraining() : Mila::Dnn::GptModel< TDeviceType, TPrecision >, Mila::Dnn::LlamaModel< TDeviceType, TPrecision >, Mila::Dnn::Model< TDeviceType, TPrecision >
• onTrainingChanging() : Mila::Dnn::FusedComponent< TDeviceType, TPrecision >, Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TLogits, TTargets, TPrecision >
• onTrainingModeChanging() : Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >, Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, 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::Lpe< TDeviceType, TIndex, 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::Swiglu< TDeviceType, TPrecision >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
• open() : Mila::Dnn::Serialization::Serializer, Mila::Dnn::Serialization::ZipSerializer
• OperationRegistry() : Mila::Dnn::Compute::OperationRegistry
• OperationsRegistrar() : Mila::Dnn::Compute::OperationsRegistrar
• operator()() : Mila::Data::BpeVocabulary::PairHash, Mila::Data::BpeVocabulary::PairViewHash, Mila::Dnn::BufferedTokenStreamer< Sink, BufSize >, Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry::CacheKeyHash, Mila::Dnn::Compute::OperationRegistry::TypeIDHash, std::hash< Mila::Dnn::Compute::DeviceId >
• operator+=() : Mila::Dnn::MemoryStats
• operator<=>() : Mila::Dnn::Compute::DeviceId
• operator=() : Mila::Core::RandomGenerator, Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >, Mila::Data::TokenSequenceLoader< TMemoryResource >, Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >, Mila::Dnn::Compute::Cuda::CublasLtLinearPlan< TComputePrecision, TParameterPrecision >, Mila::Dnn::Compute::Cuda::CublasLtMatMulPlan< TComputePrecision >, Mila::Dnn::Compute::Cuda::Rope::CudaRopeOp< TComputePrecision >, Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry, Mila::Dnn::Compute::CudaTimer, Mila::Dnn::Compute::DeviceRegistrar, Mila::Dnn::Compute::DeviceRegistry, Mila::Dnn::Compute::ExecutionContext< DeviceType::Cpu >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Cuda >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Metal >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Vulkan >, Mila::Dnn::Compute::OperationRegistry, Mila::Dnn::Compute::OperationsRegistrar, Mila::Dnn::GptModel< TDeviceType, TPrecision >, Mila::Dnn::LanguageModel< TDeviceType, TPrecision >, Mila::Dnn::LlamaModel< TDeviceType, TPrecision >, Mila::Dnn::Model< TDeviceType, TPrecision >, Mila::Dnn::ModelConfig, Mila::Dnn::Serialization::ITensorBlob, Mila::Dnn::Serialization::ModelArchive, Mila::Dnn::Serialization::ModelArchive::ScopedScope, Mila::Dnn::Serialization::TensorBlob< MR >, Mila::Dnn::Serialization::ZipSerializer, Mila::Dnn::Tensor< TDataType, TMemoryResource >, Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >, Mila::Logging::FileSink, Mila::Logging::Logger, Mila::Profiling::NvtxRange
• operator==() : Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry::CacheKey, Mila::Dnn::Compute::OperationRegistry::TypeID, Mila::Dnn::TensorShape
• operator[]() : Mila::Dnn::Tensor< TDataType, TMemoryResource >, Mila::Dnn::TensorShape
• optimize() : Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
• outputBuffer() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• outputLayout() : Mila::Dnn::Tensor< TDataType, TMemoryResource >