Setting custom kernel for CNN in pytorch

aten/src/ATen/native/Convolution.cpp might be a good starting point to see which algorithms are being dispatched.