Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
MLP.Dispatch.ixx File Reference

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

Detailed Description

Activation dispatch helpers for MLP.