[ARM][Performance] Improve ARM CPU depthwise convolution performance (#2345)
* Add sptialpack schedule for arm cpu depthwise convolution * Supply comments.
Showing
- nnvm/src/top/nn/convolution.cc 1 addition, 3 deletionsnnvm/src/top/nn/convolution.cc
- src/relay/op/nn/convolution.cc 1 addition, 2 deletionssrc/relay/op/nn/convolution.cc
- topi/python/topi/arm_cpu/conv2d.py 83 additions, 52 deletionstopi/python/topi/arm_cpu/conv2d.py
- topi/python/topi/arm_cpu/depthwise_conv2d.py 277 additions, 3 deletionstopi/python/topi/arm_cpu/depthwise_conv2d.py
Loading
Please register or sign in to comment