~/jendhub

def ud_row

bendtorch@1.3.0.0 · main.bend:2890

import bendtorch@1.3.0.0/main.bend as Bendtorch
import Base

2890# one sequence's units into rows r0..: dQ at columns j·gw (w wide), dK and dV at j·hd (kvw wide)
2891def ud_row(left: Nat, +j: Nat, +t: Nat, +gw: Nat, +hd: Nat, +w: Nat, +kvw: Nat, +r0: Nat, r: UD) -> UD:
2892  match left r:
2893    case 1n+n UD{Con{DB{bq, bk, bv}, rest}, D3{dq, dk, dv}}:
2894      ud_row(n, 1n+j, t, gw, hd, w, kvw, r0, UD{rest, D3{block_in(t, gw, w, r0, Nat.mul(j, gw), bq, dq), block_in(t, hd, kvw, r0, Nat.mul(j, hd), bk, dk), block_in(t, hd, kvw, r0, Nat.mul(j, hd), bv, dv)}})
2895    case _ r:
2896      r

Uses (7)

Used by (1)