[mlir][tosa] Added more shape inference for tosa ops
authorRob Suderman <rob.suderman@gmail.com>
Sat, 3 Jul 2021 00:00:20 +0000 (17:00 -0700)
committerRob Suderman <rob.suderman@gmail.com>
Mon, 12 Jul 2021 17:04:49 +0000 (10:04 -0700)
commit5a4e7760101581c394aa4235c55885ec960c8b3b
tree6c7b3bdd63aa572fe55eb1ee88c84b732a82d121
parenta95f56f60089ce810dd1d66a640042133f56ccd8
[mlir][tosa] Added more shape inference for tosa ops

Added shape inference for:
- scatter
- gather
- transpose
- slice
- pad
- concat
- reduction operations

Also updated reshape for more aggressive shape inference.

Differential Revision: https://reviews.llvm.org/D105383
mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
mlir/test/Dialect/Tosa/tosa-infer-shapes.mlir [new file with mode: 0644]
mlir/test/Dialect/Tosa/tosa_infer_shapes.mlir [deleted file]