Here is a list of all documented class members with links to the class documentation for each member:

A

B

C

D

E

F

G

H

I

K

L

M

N

O

P

R

S

T

U

V

W

Z

~

- r -

• rank() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• rawData() : Mila::Dnn::ITensor, Mila::Dnn::Tensor< TDataType, TMemoryResource >
• readBlob() : Mila::Dnn::Serialization::ModelArchive
• readBlobInto() : Mila::Dnn::Serialization::ModelArchive
• readMetadata() : Mila::Dnn::Serialization::ModelArchive
• readTensorBlob() : Mila::Dnn::Serialization::PretrainedModelReader
• records_removed : Mila::Distribution::RemovalReport
• registerDevice() : Mila::Dnn::Compute::DeviceRegistry
• registerDevices() : Mila::Dnn::Compute::CpuDeviceRegistrar, Mila::Dnn::Compute::CudaDeviceRegistrar, Mila::Dnn::Compute::MetalDevicePlugin
• remove() : Mila::Distribution::ModelStore
• removeComponent() : Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
• rename() : Mila::Distribution::ModelStore
• replace : Mila::Distribution::InstallOptions, Mila::Distribution::PackageRequest
• required_extension() : Mila::Dnn::VulkanTensorTraits
• requires_extensions() : Mila::Dnn::VulkanTensorTraits
• requireSerializableParameters() : Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
• reseed() : Mila::Dnn::TokenSampler< TDeviceType, TPrecision >
• reset() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >, Mila::Data::TokenSequenceLoader< TMemoryResource >, Mila::Dnn::Compute::MemoryAllocationStats
• resetKVCache() : Mila::Dnn::GemmaBlock< TDeviceType, TPrecision, kGlobal, TWeightQuant, TKvPolicy >, Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::IDecoderLayer< TDeviceType, TPrecision >
• resetKvCache() : Mila::Dnn::Compute::IKvCacheLifecycle
• reshape() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• Residual() : Mila::Dnn::Residual< TDeviceType, TPrecision >
• resize() : Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• resolveRotatedShapes() : Mila::Dnn::Rope< TDeviceType, TPrecision >
• resume_from : Mila::Distribution::HttpRequest
• retained : Mila::Distribution::RemovalReport
• revision : Mila::Distribution::HubModel, Mila::Distribution::ModelCoordinate, Mila::Distribution::ModelRecord
• rewindKvCache() : Mila::Dnn::Compute::IKvCacheLifecycle, Mila::Dnn::GemmaBlock< TDeviceType, TPrecision, kGlobal, TWeightQuant, TKvPolicy >, Mila::Dnn::GemmaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::IDecoderLayer< TDeviceType, TPrecision >, Mila::Dnn::LanguageNetwork< TDeviceType, TPrecision >
• RmsNormConfig() : Mila::Dnn::RmsNormConfig
• Rocm() : Mila::Dnn::Compute::Device
• ropeCacheBytes() : Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >
• RopeConfig() : Mila::Dnn::RopeConfig