Hi Rahan, it is a bit hard to see what is wrong due to the formatting. I call the function like this:
conv2d_cudnn.backward(module.weight.shape, gradient, input_tensor,
                      module.padding, module.stride, module.dilation, 
                      module.groups, True, False)