def head_f2

bendtorch@1.3.0.0 · main.bend:2566

import bendtorch@1.3.0.0/main.bend as Bendtorch

import Base

2566def head_f2(+t: Nat, +hd: Nat, +w: Nat, +kvw: Nat, +hc: U32, +kc: U32, v: Array<F32>, o: Array<F32>, r: QK) -> FS:
2567  match r:
2568    case QK{q, k, p}:
2569      fs_fin(q, k, pmul(Nat.sub(Nat.mul(t, hd), 1n), 0, U32.from_nat(hd), t, U32.from_nat(kvw), kc, U32.from_nat(w), hc, o, dot0(t, 0, kc, 1, U32.from_nat(kvw), p, v)))

Uses (12)

Used by (2)