def Flat.layer_norm_rows

bend-tensors@0.0.0.2 · bend_tensors.bend:2041

import Base

2041def Flat.layer_norm_rows(+rows: Nat, +cols: Nat, +eps: F32, m: Flat<rows, cols>) -> Flat<rows, cols>:
2042  FlatS{a} = m
2043  FlatS{Flat.layer_norm.go(rows, cols, U32.from_nat(cols), 0, eps, a)}