def pk_g5

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

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

889def pk_g5(+v0: F32, +v1: F32, +v2: F32, +v3: F32, +v4: F32, +q0: U32, +lim: U32, +idx: U32, +st: U32, +o: U32, t: Array<T8>, r: Array<F32> & F32) -> PK:
890  match r:
891    case Tuple{a, +v}:
892      pk_g6(v0, v1, v2, v3, v4, pick0(U32.is_lt((q0 + 5 : U32), lim), v), q0, lim, idx, st, o, t, Array.get(F32, a, (idx + (st * 6 : U32) : U32)))

Uses (13)

Used by (1)