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

- o -

• offset : Mila::Dnn::Serialization::TensorBlobMetadata
• 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 >
• op_type : Mila::Dnn::Compute::LinearOpTypeMap< DeviceType::Cpu, TensorDataType::FP32 >
• open() : Mila::Dnn::Serialization::Serializer, Mila::Dnn::Serialization::ZipSerializer
• operation_ : Mila::Dnn::Dropout< TDeviceType, TInput, TOutput >, Mila::Dnn::FusedComponent< 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::SoftmaxCrossEntropy< TDeviceType, TLogits, TTargets, TPrecision >, Mila::Dnn::Swiglu< TDeviceType, TPrecision >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
• OperationBase : Mila::Dnn::Compute::CpuCrossEntropyOp, Mila::Dnn::Compute::CpuEncoderOp
• OperationRegistry() : Mila::Dnn::Compute::OperationRegistry
• OperationsRegistrar() : Mila::Dnn::Compute::OperationsRegistrar
• OperationType : Mila::Dnn::FusedComponent< TDeviceType, TPrecision >
• 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::Component< TDeviceType, TPrecision >
• 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 >
• optimized_for_transfer : Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
• OptimizerType : Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
• OpType : Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, 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 >
• original_components_ : Mila::Dnn::FusedComponent< TDeviceType, TPrecision >
• out_features_ : Mila::Dnn::Compute::CpuLinearOp, Mila::Dnn::Compute::Cuda::Linear::CudaLinearOp< TComputePrecision, TWeightQuant >
• out_proj_ : Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• outer_shape : Mila::Dnn::MultiAxisPartition
• outer_shape_ : Mila::Dnn::RmsNorm< TDeviceType, TPrecision >
• outer_size : Mila::Dnn::AxisPartition, Mila::Dnn::MultiAxisPartition
• outer_size_ : Mila::Dnn::Compute::CpuLayerNormOp, Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >
• output_ : 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::Residual< TDeviceType, TPrecision >, Mila::Dnn::RmsNorm< TDeviceType, TPrecision >, Mila::Dnn::Swiglu< TDeviceType, TPrecision >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision >
• output_features_ : Mila::Dnn::LinearConfig
• output_shape_ : Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
• output_state_ : Mila::Dnn::Dropout< TDeviceType, TInput, TOutput >
• output_view_ : Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::LayerNorm< TDeviceType, TPrecision >, Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >, Mila::Dnn::Residual< TDeviceType, TPrecision >, Mila::Dnn::RmsNorm< TDeviceType, TPrecision >, Mila::Dnn::Swiglu< TDeviceType, TPrecision >
• outputBuffer() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• outputLayout() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• owned_context_ : Mila::Dnn::GptTransformer< TDeviceType, TPrecision >
• owned_decode_output_ : Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
• owned_exec_context_ : Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::LayerNorm< TDeviceType, TPrecision >, Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >, Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >, Mila::Dnn::MLP< 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 >
• owned_input_grad_ : Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
• owned_K_grad_ : Mila::Dnn::Rope< TDeviceType, TPrecision >
• owned_output_ : Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
• owned_Q_grad_ : Mila::Dnn::Rope< TDeviceType, TPrecision >
• owns_cache_ : Mila::Dnn::Compute::Cuda::Rope::CudaRopeOp< TComputePrecision >