Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Dnn.Component Module Reference

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.