Mila::Dnn::Compute::LinearOpConcept Concept Reference

Contract for LinearOp: typed forward matmul and backward weight/input gradients. More...

Concept definition

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

Detailed Description

Contract for LinearOp: typed forward matmul and backward weight/input gradients.

Template Parameters

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