~/jendhub

def tree4

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

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

507def tree4(~E: Data, ~f: @+t: U32 -> @+l: U32 -> @+e: E -> Array<F32> -> Array<F32> -> Array<F32> -> Array<F32> -> O4 -> O4, +d: Nat, +t: U32, +l: U32, +e: E, +x1: Array<F32>, +x2: Array<F32>, +x3: Array<F32>, +x4: Array<F32>, +oa: Array<F32>, +ob: Array<F32>, +oc: Array<F32>, +od: Array<F32>) -> O4:
508  match d:
509    case 0n: f(t, l, e, x1, x2, x3, x4, O4{oa, ob, oc, od})
510    case 1n+q:
511      x y = tree4(~E, ~f, q, (t * 2 : U32), l, e, x1, x2, x3, x4, oa, ob, oc, od) tree4(~E, ~f, q, ((t * 2 : U32) + 1 : U32), l, e, x1, x2, x3, x4, oa, ob, oc, od)
512      join4(x, y)
513
514@unsafe

Uses (8)

Used by (1)