~/jendhub

def dot_dw

bendtorch@1.3.0.0 · dtype.bend:586

import bendtorch@1.3.0.0/dtype.bend as Dtype
import Base

586def dot_dw(k: Nat, +i: U32, acc: DW, ra: Array<F32> & F32, rb: Array<F32> & F32) -> AF2:
587  match k ra rb:
588    case 0n Tuple{a, x} Tuple{b, y}:
589      AF2{a, b, DW.value(acc)}
590    case 1n+p Tuple{a, +x} Tuple{b, +y}:
591      dot_dw(p, (i + 1 : U32), DW.fma(acc, x, y), Array.get(F32, a, (i + 1 : U32)), Array.get(F32, b, (i + 1 : U32)))

Uses (12)

Used by (1)