[GitHub] [incubator-tvm] jackwish opened a new pull request #3859: [WIP] [AutoTVM][TOPI] NHWC conv2d templates for ARM

[GitBox <gi...@apache.org>]

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:
users@infra.apache.org


With regards,
Apache Git Services