def b1_if

bendtorch-gpu@1.3.0.0 · main.bend:819

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

819def b1_if(ok: Bool, +e: U32, +qa: U32, +ka: U32, +hd: Nat, +t32: U32, +sc: F32, s: S6) -> S6:
820  match ok s:
821    case True{} S6{q, k, v, d, p, g}: b1_p(e, sc, qa, ka, hd, t32, v, d, p, g, T.dot0(hd, qa, ka, 1, t32, q, k))
822    case False{} S6{q, k, v, d, p, g}: S6{q, k, v, d, p, g}

Uses (7)

Used by (2)