This op performs matrix-matrix-transpose multiplication of 4-d inputs as the following:
C[m1, n1, m0, n0] = sum_{k1, k0}(A[m1, k1, m0, k0] * B[n1, k1, n0, k0])
Paths
| Differential D105244
Add linalg.mmt4d named op ClosedPublic Authored by asaadaldien on Jun 30 2021, 4:06 PM.
Details
Summary This op performs matrix-matrix-transpose multiplication of 4-d inputs as the following: C[m1, n1, m0, n0] = sum_{k1, k0}(A[m1, k1, m0, k0] * B[n1, k1, n0, k0])
Diff Detail
Event TimelineThis revision is now accepted and ready to land.Jun 30 2021, 4:38 PM Closed by commit rG0516f49c0815: Add linalg.mmt4d named op (authored by asaadaldien). · Explain WhyJul 1 2021, 12:41 PM This revision was automatically updated to reflect the committed changes.
Revision Contents
Diff 355978 mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml
mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
|