zym1119 / DepthwiseConvolution_NVCaffe

A depthwise convolution implementation for NV_Caffe

DepthwiseConvolution_NVCaffe

A depthwise convolution implementation just for NV_Caffe

edit from sp2823(https://github.com/BVLC/caffe/pull/5665/files).

How to use?

1. copy conv_dw_layer.cpp and conv_dw_layer.cu into path_to_your_caffe/src/caffe/layers/

2. copy conv_dw_layer.hpp into path_to_your_caffe/include/caffe/layers/

3. remake your caffe