[MLIR][Shape] Allow `shape.reduce` to operate on extent tensors
authorFrederik Gossen <frgossen@google.com>
Thu, 16 Jul 2020 13:52:42 +0000 (13:52 +0000)
committerFrederik Gossen <frgossen@google.com>
Thu, 16 Jul 2020 13:53:37 +0000 (13:53 +0000)
commit0eb50e614c65d189a3f1bdf747be973829046bc1
tree7ae5c655b16ccb4ef12ef6b2301fb33972517f06
parent920e127e02531640204be0944a37bd3bf449f66b
[MLIR][Shape] Allow `shape.reduce` to operate on extent tensors

Allow `shape.reduce` to take both `shape.shape` and `tensor<?xindex>` as an
argument.

Differential Revision: https://reviews.llvm.org/D83943
mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td
mlir/lib/Dialect/Shape/IR/Shape.cpp
mlir/test/Conversion/ShapeToSCF/shape-to-scf.mlir
mlir/test/Dialect/Shape/invalid.mlir
mlir/test/Dialect/Shape/ops.mlir
mlir/test/Dialect/Shape/shape-to-shape.mlir