[TOPI] Improve conv2d for resnet18 workload (#427)
* 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
Showing
- topi/python/topi/cuda/conv2d_nchw.py 225 additions, 61 deletionstopi/python/topi/cuda/conv2d_nchw.py
- topi/python/topi/cuda/elemwise.py 0 additions, 1 deletiontopi/python/topi/cuda/elemwise.py
- topi/tests/python/test_topi_conv2d_nchw.py 1 addition, 0 deletionstopi/tests/python/test_topi_conv2d_nchw.py
Loading
Please register or sign in to comment