[mlir][Shape] Canonicalize assume_all with one input and tensor_cast of constant_shape
authorBenjamin Kramer <benny.kra@googlemail.com>
Tue, 8 Dec 2020 14:37:32 +0000 (15:37 +0100)
committerBenjamin Kramer <benny.kra@googlemail.com>
Tue, 8 Dec 2020 16:07:24 +0000 (17:07 +0100)
commit5844bc540cafb4330e7625b83371f1dab90528c3
tree46a968b4a538e3e171de972c1fea4d7bbe83e5b1
parentfebe75032f6f8322cce1dcbba11a44559aaa14e3
[mlir][Shape] Canonicalize assume_all with one input and tensor_cast of constant_shape

This allows simplifying some more complicated shape expressions

Differential Revision: https://reviews.llvm.org/D92843
mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td
mlir/lib/Dialect/Shape/IR/Shape.cpp
mlir/lib/Dialect/Shape/IR/ShapeCanonicalization.td
mlir/test/Dialect/Shape/canonicalize.mlir