def Flat.layer_norm.row2
import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensorsimport Base
2022def Flat.layer_norm.row2(+n: Nat, +p: Nat, +base: U32, +eps: F32, +cnt: F32, r: Array<F32> & F32) -> Array<F32>:
2023 (a, s) = r
2024 +mean = F32.div(s, cnt)
2025 Flat.layer_norm.row3(n, base, mean, eps, cnt, Flat.rowsq.go(p, base, mean, Array.get(F32, a, base), 0.0))