conv2d_avx_1x1.py
8.22 KB
-
AVX schedule for conv_NCHW[x]c (#1143) · d53affdc
* add conv2d_NCHWc compute template * add conv_NCHWc compute decl and schedules * allow default avx schedule * fix lint * remove unused schedule * remove import nnvm.reg * pass schedule object to compute and schedule
Yizhi Liu committed