|
Mila
Deep Neural Network Library
|
Base component interface for Mila DNN components. More...
#include <string>#include <memory>#include <ostream>#include <vector>#include <mutex>#include <atomic>#include <stdexcept>#include <format>import Serialization.SafeTensors;import Compute.DeviceTypeTraits;import Compute.IExecutionContext;import Dnn.Component:TrainingMode;import Compute.DeviceType;import Dnn.TensorDataTypeTraits;import Dnn.ITensor;import Dnn.ComponentType;import Dnn.TensorDataType;import Serialization.Mode;import Compute.DeviceId;import Serialization.ModelArchive;import Compute.CpuMemoryResource;import Dnn.TensorTypes;import Serialization.Tensor;import Dnn.Tensor;import Compute.Device;Classes | |
| class | Mila::Dnn::Component< TDeviceType, TPrecision > |
| Abstract base class for neural network components. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |