Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::OperationTraits< TOp, TDeviceType, TPrecision, TPolicy > Struct Template Referenceexport

Primary traits template for unified compile-time operation dispatch. More...

Detailed Description

template<OperationType TOp, DeviceType TDeviceType, TensorDataType TPrecision, typename TPolicy = void>
struct Mila::Dnn::Compute::OperationTraits< TOp, TDeviceType, TPrecision, TPolicy >

Primary traits template for unified compile-time operation dispatch.

Each specialization must provide a nested type alias naming the concrete operation class for the given (OperationType, DeviceType, precision, policy) combination. The primary template is intentionally undefined – an unspecialized instantiation is a hard compile error.

Template Parameters
TOpOperation identifier from the OperationType enum.
TDeviceTypeTarget device (Cpu, Cuda, ...).
TPrecisionCompute and activation precision.
TPolicyOptional policy type. Defaults to void for policy-free ops.
Examples
/__w/Mila/Mila/Mila/Src/Dnn/Components/Activations/Gelu/Gelu.ixx, and /__w/Mila/Mila/Mila/Src/Dnn/Components/Normalization/Softmax.ixx.

The documentation for this struct was generated from the following file: