mirror of https://github.com/llvm/torch-mlir
eaeb4011e6
This uses an approach inspired by what is done in IREE. See comments on LowerRankedShapes.cpp for how it works. The basic gist is that we have an op that creates a !shape.shape from a set of SSA values representing the extents, and then iteratively replace any op producing a !shape.shape with instances of that op. |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| E2E.cpp | ||
| LowerRankedShapes.cpp | ||
| LowerToHybridTensorMemRef.cpp | ||
| PassDetail.h | ||