def Mat.matmul_tn

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

import Base

1244def Mat.matmul_tn(+n: Nat, +k: Nat, +m: Nat, par: Nat, a: Mat<k, n>, b: Mat<k, m>) -> MMulTN<n, k, m>:
1245  match a b:
1246    case Mat{x} Mat{y}:
1247      mm_tn(n, k, m, split_depth(par, n, m, k), x, y)