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

- r -

• RandomGenerator() : Mila::Core::RandomGenerator
• rank() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• raw() : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
• rawData() : Mila::Dnn::ITensor, Mila::Dnn::Tensor< TDataType, TMemoryResource >
• read() : Mila::Data::MilaFileHeader
• readBlob() : Mila::Dnn::Serialization::ModelArchive
• readBlobInto() : Mila::Dnn::Serialization::ModelArchive
• readComponentMeta() : Mila::Dnn::ComponentFactory
• readHeader() : Mila::Dnn::Serialization::PretrainedModelReader
• readMetadata() : Mila::Dnn::Serialization::ModelArchive, Mila::Dnn::Serialization::PretrainedModelReader
• readTensorBlob() : Mila::Dnn::Serialization::PretrainedModelReader
• readTensorIndex() : Mila::Dnn::Serialization::PretrainedModelReader
• reduce_kv_grad() : Mila::Dnn::Compute::Cuda::Gqa::Detail::cuda_gqa_kernels< float >, Mila::Dnn::Compute::Cuda::Gqa::Detail::cuda_gqa_kernels< nv_bfloat16 >
• registerAllDevices() : Mila::Dnn::Compute::DeviceRegistrar
• registerBinaryOperation() : Mila::Dnn::Compute::OperationRegistry
• registerDevice() : Mila::Dnn::Compute::DeviceRegistry
• registerDevices() : Mila::Dnn::Compute::CpuDeviceRegistrar, Mila::Dnn::Compute::CudaDeviceRegistrar, Mila::Dnn::Compute::MetalDevicePlugin
• registerNetwork() : Mila::Dnn::NetworkFactory
• registerOperations() : Mila::Dnn::Compute::CpuAttentionOpRegistrar, Mila::Dnn::Compute::CpuCrossEntropyOpRegistrar, Mila::Dnn::Compute::CpuEncoderOpRegistrar, Mila::Dnn::Compute::CpuGeluOpRegistrar, Mila::Dnn::Compute::CpuLayerNormOpRegistrar, Mila::Dnn::Compute::CpuLinearOpRegistrar, Mila::Dnn::Compute::CpuResidualOpRegistrar, Mila::Dnn::Compute::CpuSoftmaxCrossEntropyOpRegistrar, Mila::Dnn::Compute::CpuSoftmaxOpRegistrar, Mila::Dnn::Compute::Cuda::Gelu::CudaGeluOpRegistrar, Mila::Dnn::Compute::Cuda::Gqa::CudaGroupedQueryAttentionOpRegistrar, Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOpRegistrar, Mila::Dnn::Compute::Cuda::Linear::CudaLinearOpRegistrar, Mila::Dnn::Compute::Cuda::Lpe::CudaLpeOpRegistrar, Mila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOpRegistrar, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOpRegistrar, Mila::Dnn::Compute::Cuda::Residual::CudaResidualOpRegistrar, Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOpRegistrar, Mila::Dnn::Compute::Cuda::Rope::CudaRopeOpRegistrar, Mila::Dnn::Compute::Cuda::Softmax::CudaSoftmaxOpRegistrar, Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOpRegistrar, Mila::Dnn::Compute::Cuda::Swiglu::CudaSwigluOpRegistrar, Mila::Dnn::Compute::Cuda::TokenEmbedding::CudaTokenEmbeddingOpRegistrar, Mila::Dnn::Compute::OperationsRegistrar, Mila::Dnn::Extensibility::IModulePlugin, MyCustomLayerPlugin
• registerPairedOperation() : Mila::Dnn::Compute::OperationRegistry
• registerUnaryOperation() : Mila::Dnn::Compute::OperationRegistry
• release() : Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry
• releaseCache() : Mila::Dnn::Compute::Cuda::Rope::CudaRopeOp< TComputePrecision >
• releaseResources() : Mila::Dnn::Compute::ExecutionContext< DeviceType::Cuda >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Metal >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Vulkan >
• removeComponent() : Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
• render() : Mila::Dnn::Visualization::LayerNormVisualizer, Mila::Dnn::Visualization::MLPVisualizer, Mila::Dnn::Visualization::ModuleVisualizer
• renderAll() : Mila::Dnn::Visualization::BlockVisualizer
• required_extension() : Mila::Dnn::VulkanTensorTraits
• requireMode() : Mila::Dnn::Serialization::ModelArchive
• requireOpen() : Mila::Dnn::Serialization::ModelArchive
• requires_extensions() : Mila::Dnn::VulkanTensorTraits
• reset() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >, Mila::Data::TokenSequenceLoader< TMemoryResource >, Mila::Dnn::Compute::MemoryStats
• resetKVCache() : Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• resetKvCache() : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >, Mila::Dnn::Compute::IKvCacheLifecycle
• reshape() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• Residual() : Mila::Dnn::Residual< TDeviceType, TPrecision >
• resize() : Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• resolveOutputView() : Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
• RmsNorm() : Mila::Dnn::RmsNorm< TDeviceType, TPrecision >
• RmsNormConfig() : Mila::Dnn::RmsNormConfig
• Rocm() : Mila::Dnn::Compute::Device
• Rope() : Mila::Dnn::Rope< TDeviceType, TPrecision >
• RopeCacheRegistry() : Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry
• RopeConfig() : Mila::Dnn::RopeConfig
• runBpeMergeLoop() : Mila::Data::BpeVocabulary