|
Mila
Deep Neural Network Library
|
Exported Modules | |
| module | Serialization.Tensor |
| module | Serialization.SafeTensors |
| module | Compute.DeviceTypeTraits |
| module | Compute.IExecutionContext |
| module | Serialization.ModelArchive |
| module | Dnn.TensorDataTypeTraits |
| module | Dnn.TensorDataType |
| module | Dnn.ComponentType |
| module | Compute.CpuMemoryResource |
| module | Compute.DeviceType |
| module | Dnn.ITensor |
| module | Serialization.Mode |
| module | Compute.Device |
| module | Dnn.Tensor |
| module | Compute.DeviceId |
| module | Dnn.TensorTypes |
Classes | |
| class | Mila::Dnn::BuildContext |
| Build-time context for Component::build(). More... | |
| class | Mila::Dnn::Component< TDeviceType, TPrecision > |
| Abstract base class for neural network components. More... | |
| struct | Mila::Dnn::MemoryStats |
| Memory allocation breakdown for a single component. More... | |
Enumerations | |
| enum class | Mila::Dnn::TrainingMode : uint8_t { Normal , Eval } |
| Runtime behavioral state for Components built with RuntimeMode::Training. More... | |
Functions | |
| template<TensorDataType TDataType> | |
| constexpr std::size_t | Mila::Dnn::storageBytes (dim_t element_count) noexcept |
| Storage bytes occupied by an element count of a given tensor data type. | |
| MemoryStats | Mila::Dnn::operator+ (MemoryStats lhs, const MemoryStats &rhs) noexcept |
| Aggregate two MemoryStats instances. | |
Files | |
| file | Mila/Src/Dnn/Core/Component.ixx |
| Base component interface for Mila DNN components. | |
| file | Mila/Src/Dnn/Core/Component.BuildContext.ixx |
| Build-time context passed to Component::build(). | |
| file | Mila/Src/Dnn/Core/Component.MemoryStats.ixx |
| Memory allocation statistics for component inspection. | |
| file | Mila/Src/Dnn/Core/Component.TrainingMode.ixx |
| Runtime behavioral mode for training-built Components. | |