def leaf

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

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

import Base

220def leaf(+e: E, +t: U32, +a: Array<F32>, +b: Array<F32>, c: Array<F32>) -> Array<F32>:
221  match e:
222    case E{+n, +m, +k, +tcols, +tiles, +arow, +sa, +sb, +bcol, +kg, +rot, +rem, +ac, +oa, +ob}:
223      leaf_if(U32.is_lt(t, tiles), e, t, tcols, a, b, c)
224
225@unsafe

Uses (6)

Used by (1)

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