def Mat.matmul_acc

bend-ml-tensor-array@0.1.19.0 · main.bend:1217

import Base

1217def Mat.matmul_acc(+n: Nat, +k: Nat, +m: Nat, par: Nat, a: Mat<n, k>, b: Mat<k, m>, c: Mat<n, m>) -> MMul<n, k, m>:
1218  match a b c:
1219    case Mat{x} Mat{y} Mat{z}:
1220      mmul_fin(n, k, m, gemm_any(True{}, split_depth(par, n, m, k), n, m, k, U32.from_nat(k), 1, U32.from_nat(m), 1, x, y, z))