[MLIR][Shape] Lower `shape.get_extent` to `extract_element` when possible
When the origin of a shape is an extent tensor the operation `get_extent` can be lowered directly to `extract_element`. This choice circumvents the necessity to materialize the shape in memory. Differential Revision: https://reviews.llvm.org/D82645
Loading
Please sign in to comment