jackwish commented on issue #3859: [WIP] [AutoTVM][TOPI] NHWC conv2d templates for ARM URL: https://github.com/apache/incubator-tvm/pull/3859#issuecomment-557031913 Rewritten schedule. For workload (input/output HW 64, input/output channel 64, kernel size 1), we now have comparable performance on Rasp 3B+ w/ AArch64 Ubuntu (~5ms). Tuning a larger workload currently. So basically this is ready for review, please have a look at @tmoreau89 @ZihengJiang . Notes: * Kernel prepack not enabled yet. We'd like to hold that until depthwise NHWC were done, and the AlterOpLayout were simplified. * Won't be used unless disable NHWC-NCHW layout transformation in AlterOpLayout. To try out this, put `return None` in AlterOpLayout. * AlterOpLayout will be updated when we can run whole MobileNet in NHWC.
---------------------------------------------------------------- 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
