https://github.com/pytorch/pytorch/blob/master/aten/src/THCUNN/generic/SpatialDepthwiseConvolution.cu#L59