llvm-project/mlir/python
Ahmed Taei 2c4f5690ab 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
2021-06-30 11:37:21 -07:00
..
mlir Add linalg.batch_matvec named op 2021-06-30 11:37:21 -07:00
.style.yapf
CMakeLists.txt
requirements.txt