~/jendhub

def head_bwd_full

bendtorch@1.3.0.0 · main.bend:3031

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

3031def head_bwd_full(+dm: Dm, +cols: HC, s: BS) -> BS:
3032  match dm cols s:
3033    case Dm{+t, +g, +hd, +w, +kvw, +sc} HC{+hc, +kc} BS{q, k, v, d, dq, dka, dva}:
3034      fhb1(t, hd, w, kvw, sc, hc, kc, v, d, dq, dka, dva, fprobs(Nat.sub(t, 1n), hd, sc, U32.from_nat(w), hc, U32.from_nat(kvw), kc, q, k))

Uses (7)

Used by (1)