Skip to content
Snippets Groups Projects
Commit d53affdc authored by Yizhi Liu's avatar Yizhi Liu Committed by Tianqi Chen
Browse files

AVX schedule for conv_NCHW[x]c (#1143) 

* 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
parent 3d74b48f
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment