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

- c -

• calculateAlignedSize() : Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• canHandle() : Mila::Dnn::Extensibility::IModulePlugin, MyCustomLayerPlugin
• charToIndex() : Mila::Data::CharVocabulary
• CharTokenizer() : Mila::Data::CharTokenizer
• CharTrainer() : Mila::Data::CharTrainer
• CharVocabulary() : Mila::Data::CharVocabulary
• CharVocabularyConfig() : Mila::Data::CharVocabularyConfig
• childCount() : Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
• className() : Mila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >
• cleanupBuffers() : Mila::Data::TokenSequenceLoader< TMemoryResource >
• clear() : Mila::Dnn::Visualization::Framebuffer
• clearComponents() : Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
• clearCorpus() : Mila::Data::BpeTrainer, Mila::Data::CharTrainer
• clearForwardCache() : Mila::Dnn::MLP< TDeviceType, TPrecision >
• clearGradients() : Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
• close() : Mila::Dnn::Serialization::ModelArchive, Mila::Dnn::Serialization::PretrainedModelReader, Mila::Dnn::Serialization::Serializer, Mila::Dnn::Serialization::ZipSerializer
• ColorLUT() : Mila::Dnn::Visualization::ColorLUT
• Component() : Mila::Dnn::Component< TDeviceType, TPrecision >
• CompositeComponent() : Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
• computeAttentionScores() : Mila::Dnn::Compute::CpuAttentionOp
• computeAxisPartitioning() : Mila::Dnn::Compute::CpuLayerNormOp
• computeFlatIndex() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• computeGradientAtt() : Mila::Dnn::Compute::CpuAttentionOp
• computeGradientK() : Mila::Dnn::Compute::CpuAttentionOp
• computeGradientPreatt() : Mila::Dnn::Compute::CpuAttentionOp
• computeGradientQ() : Mila::Dnn::Compute::CpuAttentionOp
• computeGradientV() : Mila::Dnn::Compute::CpuAttentionOp
• computeNormalizedFeatureCount() : Mila::Dnn::LayerNorm< TDeviceType, TPrecision >
• computeOutputValues() : Mila::Dnn::Compute::CpuAttentionOp
• computeRuntimePartition_() : Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >
• computeSize() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• computeStrides() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• configFromMetadata() : Mila::Dnn::GptModel< TDeviceType, TPrecision >, Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
• connectionTypeToString() : Mila::Dnn::ResidualConfig
• ConsoleSink() : Mila::Logging::ConsoleSink
• ConstantLRScheduler() : Mila::Dnn::ConstantLRScheduler
• convertToTokenSequences() : Mila::Data::BpeVocabulary
• copy() : Mila::Dnn::Compute::Cpu::TransferOps, Mila::Dnn::Compute::Cuda::TransferOps
• copyDeviceToDevice() : Mila::Dnn::Compute::Cuda::TransferOps
• copyDeviceToDeviceWithConversion() : Mila::Dnn::Compute::Cuda::TransferOps
• copyDeviceToHost() : Mila::Dnn::Compute::Cuda::TransferOps
• copyDeviceToHostWithConversion() : Mila::Dnn::Compute::Cuda::TransferOps
• copyFromBlob() : Mila::Dnn::Compute::Cpu::TransferOps, Mila::Dnn::Compute::Cuda::TransferOps
• copyFromBlobWithConversion() : Mila::Dnn::Compute::Cuda::TransferOps
• copyHostToDevice() : Mila::Dnn::Compute::Cuda::TransferOps
• copyHostToDeviceWithConversion() : Mila::Dnn::Compute::Cuda::TransferOps
• copyHostToHost() : Mila::Dnn::Compute::Cuda::TransferOps
• copyHostToHostWithConversion() : Mila::Dnn::Compute::Cuda::TransferOps
• CosineLRScheduler() : Mila::Dnn::CosineLRScheduler
• count() : Mila::Data::SpecialTokens
• countAll() : Mila::Data::SpecialTokens
• countPairs() : Mila::Data::BpeVocabulary
• Cpu() : Mila::Dnn::Compute::Device
• CpuAdamWOptimizer() : Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
• CpuAttentionOp() : Mila::Dnn::Compute::CpuAttentionOp
• CpuCrossEntropyOp() : Mila::Dnn::Compute::CpuCrossEntropyOp
• CpuDevice() : Mila::Dnn::Compute::CpuDevice
• CpuEncoderOp() : Mila::Dnn::Compute::CpuEncoderOp
• CpuGeluOp() : Mila::Dnn::Compute::CpuGeluOp
• CpuLayerNormOp() : Mila::Dnn::Compute::CpuLayerNormOp
• CpuLinearOp() : Mila::Dnn::Compute::CpuLinearOp
• CpuMemoryResource() : Mila::Dnn::Compute::CpuMemoryResource
• CpuResidualOp() : Mila::Dnn::Compute::CpuResidualOp
• CpuSoftmaxCrossEntropyOp() : Mila::Dnn::Compute::CpuSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets >
• CpuSoftmaxOp() : Mila::Dnn::Compute::CpuSoftmaxOp
• create() : Mila::Dnn::Detail::mlp_activation_impl< ActivationType::Gelu, TDeviceType, TPrecision >, Mila::Dnn::Detail::mlp_activation_impl< ActivationType::Swiglu, TDeviceType, TPrecision >, Mila::Dnn::NetworkFactory
• createBinaryOperation() : Mila::Dnn::Compute::OperationRegistry
• createConfigFromMetadata() : Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
• createFromArchive() : Mila::Dnn::ComponentFactory, MyCustomLayerPlugin
• createGraph() : Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::MLP< TDeviceType, TPrecision >
• createMemoryResource() : Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• createOperation() : Mila::Dnn::Dropout< TDeviceType, TInput, TOutput >, 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 >
• createOptimizer() : Mila::Dnn::Network< TDeviceType, TPrecision >
• createPairedOperation() : Mila::Dnn::Compute::OperationRegistry
• createUnaryOperation() : Mila::Dnn::Compute::OperationRegistry
• critical() : Mila::Logging::Logger
• CrossEntropyConfig() : Mila::Dnn::CrossEntropyConfig
• CublasLtError() : Mila::Dnn::Compute::CublasLtError
• CublasLtLinearPlan() : Mila::Dnn::Compute::Cuda::CublasLtLinearPlan< TComputePrecision, TParameterPrecision >
• CublasLtMatMulPlan() : Mila::Dnn::Compute::Cuda::CublasLtMatMulPlan< TComputePrecision >
• CublasLtPlanCache() : Mila::Dnn::Compute::Cuda::CublasLtPlanCache< TPlan >
• Cuda() : Mila::Dnn::Compute::Device
• cuda_gelu_impl() : Mila::Dnn::Compute::Cuda::Gelu::Detail::cuda_gelu_impl< float >, Mila::Dnn::Compute::Cuda::Gelu::Detail::cuda_gelu_impl< half >
• cuda_layernorm_impl() : Mila::Dnn::Compute::Cuda::LayerNorm::Detail::cuda_layernorm_impl< float >, Mila::Dnn::Compute::Cuda::LayerNorm::Detail::cuda_layernorm_impl< half >
• cuda_matmul_impl() : Mila::Dnn::Compute::Cuda::Linear::Detail::cuda_matmul_impl< float >
• cuda_mha_kernels() : Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail::cuda_mha_kernels< float >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail::cuda_mha_kernels< half >
• cuda_rmsnorm_impl() : Mila::Dnn::Compute::Cuda::RmsNorm::Detail::cuda_rmsnorm_impl< float >, Mila::Dnn::Compute::Cuda::RmsNorm::Detail::cuda_rmsnorm_impl< nv_bfloat16 >
• cuda_softmax_impl() : Mila::Dnn::Compute::Cuda::Softmax::Detail::cuda_softmax_impl< float >, Mila::Dnn::Compute::Cuda::Softmax::Detail::cuda_softmax_impl< half >
• cuda_swiglu_impl() : Mila::Dnn::Compute::Cuda::Swiglu::Detail::cuda_swiglu_impl< __nv_bfloat16 >, Mila::Dnn::Compute::Cuda::Swiglu::Detail::cuda_swiglu_impl< float >
• CudaAdamWOptimizer() : Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
• CudaBadAlloc() : Mila::Dnn::Compute::CudaBadAlloc
• CudaDevice() : Mila::Dnn::Compute::CudaDevice
• CudaDeviceMemoryResource() : Mila::Dnn::Compute::CudaDeviceMemoryResource
• CudaDeviceProps() : Mila::Dnn::Compute::CudaDeviceProps
• CudaError() : Mila::Dnn::Compute::CudaError
• CudaException() : CudaException
• CudaGeluOp() : Mila::Dnn::Compute::Cuda::Gelu::CudaGeluOp< TPrecision >
• CudaGqaOp() : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• CudaLayerNormOp() : Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >
• CudaLinearOp() : Mila::Dnn::Compute::Cuda::Linear::CudaLinearOp< TComputePrecision, TWeightQuant >
• CudaLpeOp() : Mila::Dnn::Compute::Cuda::Lpe::CudaLpeOp< TInput, TPrecision >
• CudaManagedMemoryResource() : Mila::Dnn::Compute::CudaManagedMemoryResource
• CudaMatMulBiasGeluOp() : Mila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >
• CudaMultiHeadAttentionOp() : Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >
• CudaPinnedMemoryResource() : Mila::Dnn::Compute::CudaPinnedMemoryResource
• CudaResidualOp() : Mila::Dnn::Compute::Cuda::Residual::CudaResidualOp< TInputA, TInputB, TPrecision >
• CudaRmsNormOp() : Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >
• CudaRopeOp() : Mila::Dnn::Compute::Cuda::Rope::CudaRopeOp< TComputePrecision >
• CudaSoftmaxCrossEntropyOp() : Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets >
• CudaSoftmaxOp() : Mila::Dnn::Compute::Cuda::Softmax::CudaSoftmaxOp< TPrecision >
• CudaSwigluOp() : Mila::Dnn::Compute::Cuda::Swiglu::CudaSwigluOp< TPrecision >
• CudaTimer() : Mila::Dnn::Compute::CudaTimer
• CudaTokenEmbeddingOp() : Mila::Dnn::Compute::Cuda::TokenEmbedding::CudaTokenEmbeddingOp< TInput, TPrecision >
• currentBatch() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
• currentPrefix() : Mila::Dnn::Serialization::ModelArchive