Compute.OperationTraits.Template Module Reference

Exported Modules
module	Compute.DeviceType
module	Compute.OperationType
module	Dnn.TensorDataType

Concepts
concept	Mila::Dnn::Compute::GqaOpConcept
	Contract for GroupedQueryAttentionOp: positional forward and backward.
concept	Mila::Dnn::Compute::LinearOpConcept
	Contract for LinearOp: typed forward matmul and backward weight/input gradients.
concept	Mila::Dnn::Compute::SamplingOpConcept
	Contract for SamplingOp: in-place token sampling from a logits tensor.
concept	Mila::Dnn::Compute::UnaryOpConcept
	Contract for policy-free unary ops (Softmax, RmsNorm, LayerNorm, Residual, ...).

Classes
struct	Mila::Dnn::Compute::OperationTraits< TOp, TDeviceType, TPrecision, TPolicy >
	Primary traits template for unified compile-time operation dispatch. More...

Files
file	/__w/Mila/Mila/Mila/Src/Dnn/Compute/Operations/OperationTraits.Template.ixx
	Unified compile-time dispatch template mapping (OperationType, DeviceType, TPrecision, TPolicy) to a concrete operation type.