I am implementing a new type of convolution layer. Therefore, we need to rewrite the forward and backward functions. However, if it is similar to an existing convolution operation, I will try to accelerate the part using cuDNN.
Now, I am referring to ConvND in torch 0.1.12 (Link 2),
and this function is being used for torch._C._cudnn_convolution. If you have any other code to refer to, can you recommend it?
Thanks!