def matmul_precise

bend-ml-tensor-fork@0.1.6.0 · main.bend:308

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

308def matmul_precise(+n: Nat, +k: Nat, +m: Nat, +par: Nat, a: TA.Mat<n, k>, b: TA.Mat<k, m>) -> TA.MMul<n, k, m>:
309  mm_precise2(Nat.is_lt(1024n, k), n, k, m, par, a, b)