def gdw_if

bendtorch-gpu@1.3.0.0 · main.bend:533

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

533def gdw_if(ok: Bool, +t: U32, +n: Nat, +hw: Nat, +hwc: U32, +c: U32, +ng: U32, +cg: U32, x: Array<F32>, dy: Array<F32>, mr: Array<F32>, ga: Array<F32>, gb: Array<F32>) -> TG.O2:
534  match ok:
535    case True{}: gdw_put(t, ga, gb, gdw_imgs(Nat.sub(n, 1n), t, U32.div(t, cg), hw, hwc, c, ng, GS{x, dy, mr, 0.0, 0.0}))
536    case False{}: TG.O2{ga, gb}