[mlir][sparse] accept sparse reshape (expand/collapse)
authorAart Bik <ajcbik@google.com>
Tue, 21 Jun 2022 23:47:44 +0000 (16:47 -0700)
committerAart Bik <ajcbik@google.com>
Wed, 22 Jun 2022 16:40:38 +0000 (09:40 -0700)
commite7d3ba1066c825c07b71deae512a40731769a963
tree840e01af1cc0be78c9cff4724178ae6a1ec9cc90
parent628a2c14e3e94442672bb3b07cd42ec9cca04531
[mlir][sparse] accept sparse reshape (expand/collapse)

This revision makes sure we accept sparse tensors as arguments
of the expand/collapse reshaping operations in the tensor dialect.
Note that the actual lowering to runnable IR is still TBD.

Reviewed By: springerm

Differential Revision: https://reviews.llvm.org/D128311
mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
mlir/test/Dialect/SparseTensor/sparse_reshape.mlir [new file with mode: 0644]