commit | e0e1bd43bff732514816958eb27c17bf77f9a25c | [log] [tgz] |
---|---|---|
author | Gaurav Shukla <gaurav@nod-labs.com> | Sun Apr 21 17:07:02 2024 +0530 |
committer | Copybara-Service <copybara-worker@google.com> | Sun Apr 21 04:43:39 2024 -0700 |
tree | c791a7a26a6408f1753a95bbd636655e4d53dd23 | |
parent | 6ca4b3b783cf01af43a014e7e55f3d410aa28f2b [diff] |
[MLIR] Generalize expand_shape to take shape as explicit input (#69267) This patch generalizes tensor.expand_shape and memref.expand_shape to consume the output shape as a list of SSA values. This enables us to implement generic reshape operations with dynamic shapes using collapse_shape/expand_shape pairs. The output_shape input to expand_shape follows the static/dynamic representation that's also used in `tensor.extract_slice`. Differential Revision: https://reviews.llvm.org/D140821 Co-authored-by: Ramiro Leal-Cavazos <ramiroleal050@gmail.com> GitOrigin-RevId: e095d978ba476c9624b4e72776089ea7301fa657
See https://mlir.llvm.org/ for more information.