def pk_g2

bend-ml-tensor-array@0.1.19.0 · main.bend:904

import bend-ml-tensor-array@0.1.19.0/main.bend as BendMlTensorArray
import Base

904def pk_g2(+v0: F32, +v1: F32, +q0: U32, +lim: U32, +idx: U32, +st: U32, +o: U32, t: Array<T8>, r: Array<F32> & F32) -> PK:
905  match r:
906    case Tuple{a, +v}:
907      pk_g3(v0, v1, pick0(U32.is_lt((q0 + 2 : U32), lim), v), q0, lim, idx, st, o, t, Array.get(F32, a, (idx + (st * 3 : U32) : U32)))

Uses (13)

Used by (1)