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

- r -

• r : Mila::Dnn::Visualization::RGB
• 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 >
• ref_count : Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry::CacheEntry
• 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
• registered_ids_ : Mila::Dnn::Compute::DeviceRegistry
• 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
• RegistryMap : 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_device_type : Mila::Dnn::TensorDataTypeMap< __nv_fp8_e4m3 >, Mila::Dnn::TensorDataTypeMap< __nv_fp8_e5m2 >, Mila::Dnn::TensorDataTypeMap< half >, Mila::Dnn::TensorDataTypeMap< nv_bfloat16 >
• required_extension() : Mila::Dnn::VulkanTensorTraits
• requireMode() : Mila::Dnn::Serialization::ModelArchive
• requireOpen() : Mila::Dnn::Serialization::ModelArchive
• requires_context_binding : Mila::Dnn::Compute::MemoryResourceTraits< CudaDeviceMemoryResource >, Mila::Dnn::Compute::MemoryResourceTraits< CudaManagedMemoryResource >, Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
• requires_extensions() : Mila::Dnn::VulkanTensorTraits
• requires_host_synchronization : Mila::Dnn::Compute::MemoryResourceTraits< CpuMemoryResource >, Mila::Dnn::Compute::MemoryResourceTraits< CudaDeviceMemoryResource >, Mila::Dnn::Compute::MemoryResourceTraits< CudaManagedMemoryResource >, Mila::Dnn::Compute::MemoryResourceTraits< CudaPinnedMemoryResource >
• res1_ : Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• res1_prefill_ : Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• res2_ : Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• 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 >
• residual_norms : Mila::Dnn::Visualization::VisualizerContext
• residual_scale_ : Mila::Dnn::GptBlockConfig
• ResidualType : Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• resize() : Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• resolveOutputView() : Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
• rms1_ : Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• rms2_ : Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• rms_norm_eps_ : Mila::Dnn::LlamaConfig
• RmsNorm() : Mila::Dnn::RmsNorm< TDeviceType, TPrecision >
• RmsNormConfig() : Mila::Dnn::RmsNormConfig
• RmsNormType : Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
• rng_ : Mila::Dnn::Dropout< TDeviceType, TInput, TOutput >
• Rocm() : Mila::Dnn::Compute::Device
• Rope() : Mila::Dnn::Rope< TDeviceType, TPrecision >
• rope_ : Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• rope_scaling_factor_ : Mila::Dnn::LlamaConfig
• rope_theta : Mila::Dnn::Serialization::PretrainedMetadata
• rope_theta_ : Mila::Dnn::LlamaConfig
• RopeCacheRegistry() : Mila::Dnn::Compute::Cuda::Rope::RopeCacheRegistry
• RopeConfig() : Mila::Dnn::RopeConfig
• RopeType : Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• rotary_dim_ : Mila::Dnn::RopeConfig
• rstd_ : Mila::Dnn::Compute::CpuLayerNormOp, Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >, Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >
• rstd_tensor_ : Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >, Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >
• runBpeMergeLoop() : Mila::Data::BpeVocabulary
• runtime_mode_ : Mila::Dnn::BuildContext, Mila::Dnn::Model< TDeviceType, TPrecision >