Mila::Dnn::Compute::UnaryOpConcept Concept Reference

Contract for policy-free unary ops (Softmax, RmsNorm, LayerNorm, Residual, ...). More...

Concept definition

template<typename TOp, typename TTensor>
concept Mila::Dnn::Compute::UnaryOpConcept =  requires( const TOp& op, const TTensor& in, TTensor& out )
    {
        op.forward( in, out );
    }

Detailed Description

Contract for policy-free unary ops (Softmax, RmsNorm, LayerNorm, Residual, ...).

Template Parameters

TOp	Candidate op type.
TTensor	Tensor type at the call site.