def Mat.matmul

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

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

1208def Mat.matmul(+n: Nat, +k: Nat, +m: Nat, par: Nat, a: Mat<n, k>, b: Mat<k, m>) -> MMul<n, k, m>:
1209  match a b:
1210    case Mat{x} Mat{y}:
1211      mm_a(n, k, m, split_depth(par, n, m, k), x, y)

Uses (5)

Used by (1)