|
Mila 0.13.48
Deep Neural Network Library
|
#include <memory>#include <vector>#include <string>#include <math.h>#include <iostream>#include <stdexcept>import Compute.CpuMemoryResource;import Compute.OperationRegistry;import Dnn.Tensor;import Compute.DeviceType;import Dnn.ITensor;import Compute.OperationType;import Dnn.TensorTypes;import Compute.UnaryOperation;import Dnn.TensorDataType;import Dnn.Components.GeluConfig;import Dnn.ComponentConfig;import Compute.ExecutionContext;import Compute.ExecutionContextTemplate;import Compute.IExecutionContext;import Dnn.Component;Classes | |
| class | Mila::Dnn::Compute::CpuGeluOp |
| CPU implementation of GELU activation operation using abstract TensorDataType. More... | |
| class | Mila::Dnn::Compute::CpuGeluOpRegistrar |
| Class responsible for registering CPU GELU operations. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
Macros | |
| #define | _USE_MATH_DEFINES |
Variables | |
| constexpr float | Mila::Dnn::Compute::GELU_SCALING_FACTOR = 0.7978845608f |
| Scaling factor for GELU tanh approximation: sqrt(2/pi). | |
| #define _USE_MATH_DEFINES |