~/jendhub

def fg_leaf

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

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

731# leaf t writes the pairs t, t + l, .. of the slice's pairs, each into the cells of it that lie in the slice
732def fg_leaf(+t: U32, +l: U32, +e: FE, x1: Array<F32>, x2: Array<F32>, x3: Array<F32>, x4: Array<F32>, o: Array<F32>) -> Array<F32>:
733  match e:
734    case FE{+s, +i0, +n}: fg_go(share(t, fg_pairs(i0, n), l), t, l, s, U32.shrn(i0, 1n), U32.and(i0, 1), n, o)

Uses (9)

Used by (1)