~/jendhub

def mm_precise2

bend-ml-tensor-gpu@0.1.5.1 · main.bend:565

import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpu
import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

565def mm_precise2(long: Bool, +n: Nat, +k: Nat, +m: Nat, +par: Nat, a: TA.Mat<n, k>, b: TA.Mat<k, m>) -> TA.MMul<n, k, m>:
566  match long a b:
567    case True{} TA.Mat{x} TA.Mat{y}: mm_precise3(n, k, m, x, y)
568    case False{} a b: matmul(n, k, m, par, a, b)

Uses (6)

Used by (1)