BendMlTensorGpu.fu_leaf: def of bend-ml-tensor-gpu@0.1.5.1 Import: import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpu Lines 705-708 of bend-ml-tensor-gpu@0.1.5.1/main.bend: https://jend.leeeet.dev/read.txt?path=bend-ml-tensor-gpu@0.1.5.1/main.bend&start_line=705&end_line=708 import Base # leaf t writes the cells t, t + l, .. below n, cell k the uniform of index i0 + k def fu_leaf(+t: U32, +l: U32, +e: FE, x1: Array, x2: Array, x3: Array, x4: Array, o: Array) -> Array: match e: case FE{+s, +i0, +n}: fu_go(share(t, n, l), t, l, s, i0, o) Uses (6): https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Array https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32 https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32 https://jend.leeeet.dev/definition.txt?path=bend-ml-tensor-gpu@0.1.5.1/main.bend&name=FE https://jend.leeeet.dev/definition.txt?path=bend-ml-tensor-gpu@0.1.5.1/main.bend&name=fu_go https://jend.leeeet.dev/definition.txt?path=bend-ml-tensor-gpu@0.1.5.1/main.bend&name=share