def Flat.layer_norm.row2

bend-tensors@0.0.0.2 · bend_tensors.bend:2022

import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
import 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))

Uses (10)

Used by (1)