def wr

bend-ml-tensor-gpu@0.1.5.1 · main.bend:193

import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpu

import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

193# writes cell (i, j) of C when it is inside n x m
194def wr(ok: Bool, +ac: Bool, +idx: U32, +x: F32, c: Array<F32>) -> Array<F32>:
195  match ok:
196    case True{}: TA.cwr(ac, idx, x, c)
197    case False{}: c

Uses (5)

Used by (1)

bend-ml-tensor-gpu@0.1.5.1/main:put