~/jendhub

def tkv_sel

bendtorch-gpu@1.3.0.0 · main.bend:722

import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpu
import Base

722def tkv_sel(lo: Bool, +e: U32, +nk: U32, +t32: U32, +kv: U32, +hd32: U32, +kvw: U32, s: S4) -> S4:
723  match lo s:
724    case True{} S4{k, v, kt, vt}: tkv_k(v, vt, tk_put(e, kt, Array.get(F32, k, tk_src(e, t32, kv, hd32, kvw))))
725    case False{} S4{k, v, kt, vt}: tkv_v(k, kt, tk_put((e - nk : U32), vt, Array.get(F32, v, tk_src((e - nk : U32), t32, kv, hd32, kvw))))

Uses (10)

Used by (1)