def gnb_if

bendtorch-gpu@1.3.0.0 · main.bend:486

import Base
import bend-ml-tensor-gpu@0.1.5.0/main.bend as TG

486def gnb_if(ok: Bool, +t: U32, +ng: U32, +hw: Nat, +hwc: U32, +c: U32, +cg: Nat, +cg32: U32, +nf: F32, +eps: F32, x: Array<F32>, dy: Array<F32>, g: Array<F32>, dx: Array<F32>, mr: Array<F32>) -> TG.O2:
487  match ok:
488    case True{}: gnb_blk(hw, (U32.div(t, ng) * hwc : U32), (U32.mod(t, ng) * cg32 : U32), c, cg, nf, eps, t, x, dy, g, dx, mr)
489    case False{}: TG.O2{dx, mr}