Implementing a custom convolution using conv2d_input and conv2d_weight

26

If you are looking for the intuition behind using at::cudnn_convolution, you can look here. The exact definition of it may vary based on the version of cuDNN, for v7 look here, and for v8 here.

Here is a detailed explanation about benchmark.

1 Like