Quinn Dawkins 60e562d11a [mlir][linalg] Add unit dim folding pattern for tensor.pad (#84684) ...

Unit extent dims that are not padded by a tensor.pad can be folded away.
When folding unit extent dims of surrounding linalg ops, this increases
the chance that the iteration space of the linalg op will align with
nearby pad ops, improving fusion opportunities.

2024-03-11 18:24:23 -04:00

34 KiB

Raw Blame History

View Raw