jackwish opened a new pull request #3859: [WIP] [AutoTVM][TOPI] NHWC conv2d templates for ARM URL: https://github.com/apache/incubator-tvm/pull/3859 Per https://github.com/dmlc/tvm/pull/3754 and https://github.com/dmlc/tvm/pull/3141#issuecomment-526434398 , we are enabling NHWC conv2d templates for ARM as a *nearly* final solution. The benefits include: * Enable NHWC schedule directly. Previously, we need to transpose between NCHW and NHWC. * AutoTVM now can tune NHWC directly. Previously, we need to build a NCHW network to tune. * Potential performance advantage in NHWC which known to community. Cowork with @FrozenGene and @etaf . This is a **draft** to loop people who may have interest, @anijain2305 @tmoreau89 . Will loop more when the PR is ready, thank you. :)
---------------------------------------------------------------- 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
