|
Mila
Deep Neural Network Library
|
Implementation of the CPU-based residual operation for neural networks. More...
#include <memory>#include <vector>#include <string>#include <iostream>#include <stdexcept>#include <math.h>import Compute.CpuDevice;import Compute.CpuMemoryResource;import Compute.MemoryResource;import Compute.OperationBase;import Compute.OperationRegistry;import Compute.OperationType;import Compute.BinaryOperation;import Compute.DeviceType;import Compute.OperationAttributes;import Dnn.ComponentConfig;import Compute.DeviceContext;import Dnn.Tensor;import Dnn.Modules.Residual;Classes | |
| class | Mila::Dnn::Compute::CpuResidualOp |
| CPU implementation of the residual operation for neural networks. More... | |
| class | Mila::Dnn::Compute::CpuResidualOpRegistrar |
| Class responsible for registering the CpuResidualOp operation. More... | |
Namespaces | |
| namespace | Mila |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
Macros | |
| #define | _USE_MATH_DEFINES |
Implementation of the CPU-based residual operation for neural networks.
| #define _USE_MATH_DEFINES |