diff options
| author | Ahmed Taei <ataei@google.com> | 2021-06-22 12:50:10 -0700 |
|---|---|---|
| committer | Ahmed Taei <ataei@google.com> | 2021-06-30 11:37:21 -0700 |
| commit | 2c4f5690ab5e435691aafe554725dbbd521b3754 (patch) | |
| tree | 56d37b9f3670ad5c01c1585cf8e27cc45684ae11 /llvm/lib/CodeGen/MachineOperand.cpp | |
| parent | 03051f7ac8a3e2eda44b8280290c90863a064f0e (diff) | |
| download | llvm-2c4f5690ab5e435691aafe554725dbbd521b3754.zip llvm-2c4f5690ab5e435691aafe554725dbbd521b3754.tar.gz llvm-2c4f5690ab5e435691aafe554725dbbd521b3754.tar.bz2 | |
Add linalg.batch_matvec named op
Similarly to batch_mat vec outer most dim is a batching dim
and this op does |b| matrix-vector-products :
C[b, i] = sum_k(A[b, i, k] * B[b, k])
Reviewed By: rsuderman
Differential Revision: https://reviews.llvm.org/D104739
Diffstat (limited to 'llvm/lib/CodeGen/MachineOperand.cpp')
0 files changed, 0 insertions, 0 deletions