~/jendhub

def gdw_blk

bendtorch-gpu@1.3.0.0 · main.bend:515

import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpu
import Base

515# block bl's mean and 1/sqrt(var + eps), then its rows
516def gdw_blk(+hw: Nat, +i: U32, +bl: U32, +c: U32, s: GS) -> GS:
517  match s:
518    case GS{x, dy, mr, sg, sb}: gdw_blk1(hw, i, bl, c, sg, sb, x, dy, Array.get(F32, mr, (bl * 2 : U32)))

Uses (7)

Used by (1)