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

- a -

• acquire() : Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry
• activationChildName() : Mila::Dnn::MLP< TDeviceType, TPrecision >
• AdamWOptimizer() : Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
• add() : Mila::Dnn::Compute::Cpu::MathOps, Mila::Dnn::Compute::Cuda::MathOps
• addActivation() : Mila::Dnn::MLP< TDeviceType, TPrecision >
• addComponent() : Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
• addCorpus() : Mila::Data::TokenizerTrainer
• addCorpusFromFile() : Mila::Data::BpeTrainer, Mila::Data::CharTrainer, Mila::Data::TokenizerTrainer
• addCorpusFromStream() : Mila::Data::BpeTrainer, Mila::Data::CharTrainer, Mila::Data::TokenizerTrainer
• addData() : Mila::Dnn::Serialization::ArchiveSerializer, Mila::Dnn::Serialization::ZipSerializer
• addImpl() : Mila::Dnn::Compute::Cuda::MathOps
• addLinear() : Mila::Dnn::MLP< TDeviceType, TPrecision >
• addMetadata() : Mila::Dnn::Serialization::ArchiveSerializer, Mila::Dnn::Serialization::ModelArchive, Mila::Dnn::Serialization::ZipSerializer
• addParameter() : Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >, Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >, Mila::Dnn::Compute::Optimizer< TDeviceType, TPrecision >, Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
• addRegularTokens() : Mila::Data::CharVocabulary
• addSpecialToken() : Mila::Data::BpeVocabulary
• addSpecialTokensFromConfig() : Mila::Data::BpeVocabulary, Mila::Data::CharVocabulary
• alignedSize() : Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• allocateBuffer() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• allocateBuffers() : Mila::Data::TokenSequenceLoader< TMemoryResource >
• allocateParameters() : Mila::Dnn::RmsNorm< TDeviceType, TPrecision >
• allocateStateTensors() : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• allocateStatisticsTensors() : Mila::Dnn::Compute::CpuLayerNormOp
• applyMergeAndUpdateCounts() : Mila::Data::BpeVocabulary
• applySoftmax() : Mila::Dnn::Compute::CpuAttentionOp
• asInputA() : Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB >
• asInputB() : Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB >
• asInputTensor() : Mila::Dnn::Compute::UnaryOperation< TDeviceType, TInput, TPrecision >
• asLeftTensor() : Mila::Dnn::Compute::BinaryOperation< TDeviceType, TPrecision, TInputA, TInputB >
• asOutputTensor() : Mila::Dnn::Compute::BinaryOperation< TDeviceType, TPrecision, TInputA, TInputB >, Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB >, Mila::Dnn::Compute::UnaryOperation< TDeviceType, TInput, TPrecision >
• asRightTensor() : Mila::Dnn::Compute::BinaryOperation< TDeviceType, TPrecision, TInputA, TInputB >