def pk_g3

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

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

899def pk_g3(+v0: F32, +v1: F32, +v2: F32, +q0: U32, +lim: U32, +idx: U32, +st: U32, +o: U32, t: Array<T8>, r: Array<F32> & F32) -> PK:
900  match r:
901    case Tuple{a, +v}:
902      pk_g4(v0, v1, v2, pick0(U32.is_lt((q0 + 3 : U32), lim), v), q0, lim, idx, st, o, t, Array.get(F32, a, (idx + (st * 4 : U32) : U32)))

Uses (13)

Used by (1)