Here is a list of all module functions with links to the module documentation for each function:

- c -

• Mila::Dnn::Compute::cast_context_() : Compute.ExecutionContext
• Mila::Dnn::Compute::Cuda::ceil_div() : Cuda.Helpers
• Mila::Dnn::Compute::Cuda::checkDevice() : Cuda.Helpers
• Mila::Data::TokenSequenceLoader::cleanupBuffers() : Data.TokenSequenceLoader
• Mila::Dnn::MLP::clearForwardCache() : Dnn.Components.MLP
• Mila::Dnn::Component::Component() : Dnn.Component
• Mila::Dnn::Compute::Cuda::computeArchitectureBuckets() : Compute.CublasLtPlanCache
• Mila::Dnn::computeAxisPartition() : Dnn.TensorPartitioning
• Mila::Dnn::LayerNorm::computeNormalizedFeatureCount() : Dnn.Components.LayerNorm
• Mila::Dnn::computeNormalizedShapePartition() : Dnn.TensorPartitioning
• Mila::Dnn::computeNumElements() : Dnn.TensorPartitioning
• Mila::Dnn::computePrefillChunkSize() : Dnn.Components.LlamaTransformer
• Mila::Dnn::connectionTypeToString() : Dnn.ConnectionType
• Mila::Dnn::GptTransformer::createConfigFromMetadata() : Dnn.Components.GptTransformer
• Mila::Dnn::LlamaTransformer::createConfigFromMetadata() : Dnn.Components.LlamaTransformer
• Mila::Dnn::Compute::createExecutionContext() : Compute.ExecutionContextFactory
• Mila::Dnn::GptBlock::createGraph() : Dnn.Components.GptBlock
• Mila::Dnn::GptTransformer::createGraph() : Dnn.Components.GptTransformer
• Mila::Dnn::LlamaTransformer::createGraph() : Dnn.Components.LlamaTransformer
• Mila::Dnn::MLP::createGraph() : Dnn.Components.MLP
• Mila::Dnn::Dropout::createOperation() : Dnn.Components.Dropout
• Mila::Dnn::Gelu::createOperation() : Dnn.Components.Gelu
• Mila::Dnn::GroupedQueryAttention::createOperation() : Dnn.Components.Gqa
• Mila::Dnn::LayerNorm::createOperation() : Dnn.Components.LayerNorm
• Mila::Dnn::Linear::createOperation() : Dnn.Components.Linear
• Mila::Dnn::Lpe::createOperation() : Dnn.Components.Lpe
• Mila::Dnn::MultiHeadAttention::createOperation() : Dnn.Components.MultiHeadAttention
• Mila::Dnn::Residual::createOperation() : Dnn.Components.Residual
• Mila::Dnn::RmsNorm::createOperation() : Dnn.Components.RmsNorm
• Mila::Dnn::Rope::createOperation() : Dnn.Components.Rope
• Mila::Dnn::Softmax::createOperation() : Dnn.Components.Softmax
• Mila::Dnn::SoftmaxCrossEntropy::createOperation() : Dnn.Components.SoftmaxCrossEntropy
• Mila::Dnn::Swiglu::createOperation() : Dnn.Components.Swiglu
• Mila::Dnn::TokenEmbedding::createOperation() : Dnn.Components.TokenEmbedding
• Mila::Dnn::Compute::Cuda::cublaslt_compute_types() : CublasLt.Utils
• Mila::Dnn::Compute::Cuda::cublaslt_cuda_data_type() : CublasLt.Utils
• Mila::Dnn::Compute::cublasLtCheckStatus() : CublasLt.Error
• Mila::Dnn::Compute::Cuda::CublasLtMatMulPlan::CublasLtMatMulPlan() : Compute.CublasLtPlan
• Mila::Dnn::Compute::cudaCheckLastError() : Cuda.Error
• Mila::Dnn::Compute::cudaCheckStatus() : Cuda.Error