anijain2305 opened a new pull request #4600: [Relay][Convert Layout] Handling batch norm layout change. URL: https://github.com/apache/incubator-tvm/pull/4600 @yzhliu @trevor-m Convert Layout pass was earlier using SimplifyInference pass as a pre-requisite to expand batch norm to simpler ops. However, @trevor-m told that TRT offloading of Relay graph works better if we keep batch norm intact in Relay. This caused a problem with Convert Layout pass that inserts layout transforms with batch norm. This PR propagates the new layout through batch norm, preventing the insertion of layout transforms.
---------------------------------------------------------------- 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] With regards, Apache Git Services
