def hb4

bendtorch@1.3.0.0 · main.bend:2662

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

2662# dK += Gᵀ·Q, then dV += Pᵀ·dO
2663def hb4(+t: Nat, +hd: Nat, +w: Nat, +hc: U32, k: Array<F32>, d: Array<F32>, v: Array<F32>, p: Array<F32>, dq: Array<F32>, dva: Array<F32>, r: PO) -> BS:
2664  match r:
2665    case PO{g, q, dka}:
2666      hb5(q, k, v, dq, dka, pacc(Nat.sub(Nat.mul(t, hd), 1n), 0, U32.from_nat(hd), t, U32.from_nat(w), hc, dva, dot0(t, 0, hc, U32.from_nat(t), U32.from_nat(w), p, d)))

Uses (12)

Used by (1)