|
Mila
Deep Neural Network Library
|
Device-templated Linear (fully connected) component. More...
#include <memory>#include <vector>#include <string>#include <iostream>#include <sstream>#include <type_traits>#include <stdexcept>#include <cstdint>#include <cstring>#include <format>#include <optional>import Dnn.TensorHelpers;import Dnn.TensorOps;import Serialization.Metadata;import Serialization.ModelArchive;import nlohmann.json;import Compute.OperationTraits;import Logging.Logger;import Compute.OperationType;import Compute.ExecutionContextFactory;import Serialization.SafeTensors;import Compute.DeviceTypeTraits;import Serialization.Mode;import Dnn.Component;import Dnn.Tensor;import Serialization.Tensor;import Dnn.ComponentType;import Compute.CpuMemoryResource;import Dnn.TensorDataTypeTraits;import Dnn.Components.LinearConfig;import Dnn.ITensor;import Compute.DeviceType;import Compute.IExecutionContext;import Dnn.TensorDataType;import Dnn.TensorTypes;import Dnn.Quantization.Weight.Policies;import Compute.DeviceId;import Compute.MemoryResource;import Compute.Device;Classes | |
| class | Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant > |
| Device-templated fully connected (linear) component. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
Typedefs | |
| using | Mila::Dnn::json = nlohmann::json |
Device-templated Linear (fully connected) component.