csullivan opened a new pull request #7807:
URL: https://github.com/apache/tvm/pull/7807
This PR adds support to SimplifyTranspose to simplify consecutive
transpose<->layout_transform operations when the layout transform involves a
rank expansion or contraction. The added unit test fails on main because the
current SimplifyTranspose only doesn't consider the case when the layout
transform and transpose may act on tensors of different rank. As an example,
`NCHW4c -> LayoutTransform -> NCHW -> Transpose -> NHWC`
now simplifies to
`NCHW4c -> LayoutTransform -> NHWC`.
The reverse, rank expansion, is also supported.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
