def gdw_put

bendtorch-gpu@1.3.0.0 · main.bend:526

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

526def gdw_put(+j: U32, ga: Array<F32>, gb: Array<F32>, s: GS) -> TG.O2:
527  match s:
528    case GS{x, dy, mr, +sg, +sb}: TG.O2{gdg_put(j, sg, Array.get(F32, ga, j)), gdg_put(j, sb, Array.get(F32, gb, j))}