anijain2305 edited a comment on issue #4542: [AutoTVM] Tuning fails for an NHWC network on Arm CPU URL: https://github.com/apache/incubator-tvm/issues/4542#issuecomment-569121299 I agree that AlterOpLayout and AutoTVM do not have the most intuitive design/user experience yet. For now, you can choose either of the 3 options 1) Use the #3859 to use NHWC schedule. However, it might not give you the best performance because for now NCHW perf is somewhat better than NHWC. 2) Call AlterOpLayout pass the way you did before extracting tasks. 3) In general, use the #4335 ConvertLayout to convert the whole graph to NCHW and then call AutoTVM extraction. Similarly, after AutoTVM is done, before calling the relay.build, call ConvertLayout.
---------------------------------------------------------------- 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
