def leaf_if

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

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

import Base

214def leaf_if(ok: Bool, +e: E, +t: U32, +tcols: U32, +a: Array<F32>, +b: Array<F32>, c: Array<F32>) -> Array<F32>:
215  match ok:
216    case True{}: tile(e, (U32.div(t, tcols) * 4 : U32), (U32.mod(t, tcols) * 4 : U32), a, b, c)
217    case False{}: c
218
219@unsafe

Uses (9)

Used by (1)

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