def gn_mr

bendtorch@1.3.0.0 · main.bend:1903

import Base

1903# a block's mean and 1 / sqrt(var + eps); off is its first cell, nf = hw·cg
1904def gn_mr(+hw: Nat, +off: U32, +c: U32, +cg: Nat, +nf: F32, +eps: F32, x: Array<F32>) -> ASM:
1905  gn_mr1(hw, off, c, cg, nf, eps, gn_sum(Nat.sub(hw, 1n), off, c, cg, asum(Nat.sub(cg, 1n), off, 0.0, Array.get(F32, x, off))))