[MLIR][Shape] Lower `shape_of` for unranked tensors
authorFrederik Gossen <frgossen@google.com>
Thu, 25 Jun 2020 08:50:02 +0000 (08:50 +0000)
committerFrederik Gossen <frgossen@google.com>
Thu, 25 Jun 2020 08:50:45 +0000 (08:50 +0000)
commite34b88309e7dfd86d298d01b7ab36e5858f1b62f
tree9eb81f9c366e31b2cfce3e49f6883c2033e83a2f
parent24debf5a76e03cfeab6df2530151e3061eb110f2
[MLIR][Shape] Lower `shape_of` for unranked tensors

Lower `shape_of` for unranked tensors.
Materializes shape in stack-allocated memory.

Differential Revision: https://reviews.llvm.org/D82196
mlir/include/mlir/Dialect/StandardOps/IR/Ops.td
mlir/lib/Conversion/ShapeToSCF/ShapeToSCF.cpp
mlir/lib/Dialect/StandardOps/IR/Ops.cpp
mlir/test/Conversion/ShapeToSCF/shape-to-scf.mlir