|
Mila 0.13.48
Deep Neural Network Library
|
GELU activation component implementation. More...
#include <memory>#include <vector>#include <string>#include <sstream>#include <type_traits>#include <stdexcept>#include <format>#include <utility>#include <optional>#include <algorithm>import Serialization.Metadata;import Serialization.Mode;import Serialization.Tensor;import Logging.Logger;import Compute.UnaryOperation;import Compute.ExecutionContextFactory;import Compute.IExecutionContext;import Dnn.Components.GeluConfig;import Dnn.Tensor;import Dnn.TensorDataType;import Compute.Device;import Serialization.ModelArchive;import Dnn.Component;import Dnn.TensorTypes;import Dnn.ComponentType;import Compute.CpuMemoryResource;import Dnn.ITensor;import Compute.OperationTraits;import Compute.DeviceId;import Dnn.ApproximationMethod;import Dnn.TensorDataTypeTraits;import Compute.DeviceType;import Compute.DeviceTypeTraits;Classes | |
| class | Mila::Dnn::Gelu< TDeviceType, TPrecision > |
| Gaussian Error Linear Unit (GELU) activation component. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
GELU activation component implementation.
Device-templated GELU component that delegates computation to a registered device-specific UnaryOperation backend.