|
Mila 0.13.48
Deep Neural Network Library
|
Activation dispatch helpers for MLP. More...
#include <memory>#include <functional>#include <stdexcept>#include <string>#include <optional>import Dnn.Component;import Dnn.Components.Swiglu;import Compute.DeviceType;import Dnn.Components.Gelu;import Dnn.TensorDataType;import Compute.DeviceTypeTraits;import Dnn.Tensor;import Dnn.ActivationType;Classes | |
| struct | Mila::Dnn::Detail::mlp_activation_impl< ActivationType::Gelu, TDeviceType, TPrecision > |
| struct | Mila::Dnn::Detail::mlp_activation_impl< ActivationType::Swiglu, TDeviceType, TPrecision > |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Detail |
Activation dispatch helpers for MLP.