|
Mila 0.13.48
Deep Neural Network Library
|
CPU implementation of Linear (fully connected) operation. More...
#include <memory>#include <vector>#include <string>#include <sstream>#include <iostream>#include <stdexcept>#include <cstdint>#include <type_traits>#include <optional>import Compute.CpuTensorDataTypeTraits;import Compute.MemoryResource;import Compute.OperationRegistry;import Dnn.Component;import Compute.OperationBase;import Dnn.Components.LinearConfig;import Dnn.TensorTypes;import Dnn.ITensor;import Dnn.TensorDataType;import Dnn.TensorDataTypeTraits;import Compute.CpuDevice;import Dnn.ComponentConfig;import Dnn.TensorHostTypeMap;import Compute.ExecutionContextTemplate;import Compute.DeviceType;import Compute.IExecutionContext;import Dnn.Tensor;import Compute.ExecutionContext;import Compute.CpuMemoryResource;import Compute.UnaryOperation;import Compute.OperationType;Classes | |
| class | Mila::Dnn::Compute::CpuLinearOp |
| CPU implementation of Linear operation using abstract TensorDataType API. More... | |
| class | Mila::Dnn::Compute::CpuLinearOpRegistrar |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
CPU implementation of Linear (fully connected) operation.