pub fn recognize_l2norm(stmts: &[Stmt<'_>]) -> Option<L2NormSpec>Expand description
Recognize Let Y be l2norm of X [across H heads] — the per-row sqrt denominator bound in a Let and
the X[base+q] / denom normalize loop.
pub fn recognize_l2norm(stmts: &[Stmt<'_>]) -> Option<L2NormSpec>Recognize Let Y be l2norm of X [across H heads] — the per-row sqrt denominator bound in a Let and
the X[base+q] / denom normalize loop.