test_topi_conv2d_nchw.py
2.7 KB
-
[TOPI] Improve conv2d for resnet18 workload (#427) · ab17bf65
* relu activation migrated to topi * reviews addressed * relu compute deleted * conv2d_nchw updated * resnet18 hand tuned schedule added * pylint error fixed * one more workload test for conv2d_nchw * conv2d schedule subfunctions added for different patterns * reviews addressed
Leyuan Wang committed