[mlir][shape] Use IndexElementsAttr in Shape dialect.
authorSean Silva <silvasean@google.com>
Tue, 26 May 2020 23:45:32 +0000 (16:45 -0700)
committerSean Silva <silvasean@google.com>
Wed, 27 May 2020 20:39:49 +0000 (13:39 -0700)
commit25132b36a8b39e7c2b0b28aa73772e57191b6df4
tree70f5b99fb17a49d9a4a01019d2dc52df06c47e88
parent9546d8b108dce03e03e0448cebbca5fa0fe4be21
[mlir][shape] Use IndexElementsAttr in Shape dialect.

Summary:
Index is the proper type for storing shapes when constant folding, so
this fixes the previous code (which was using i64).

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