def fu_leaf
import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpuimport Base
705# leaf t writes the cells t, t + l, .. below n, cell k the uniform of index i0 + k
706def fu_leaf(+t: U32, +l: U32, +e: FE, x1: Array<F32>, x2: Array<F32>, x3: Array<F32>, x4: Array<F32>, o: Array<F32>) -> Array<F32>:
707 match e:
708 case FE{+s, +i0, +n}: fu_go(share(t, n, l), t, l, s, i0, o)