Hi, I just find the conv2d forward code but do not know where to set its backward function.
aten/src/ATen/native/Convolution.cpp is a good starting point, but depending on the inputs/device etc, it will dispatch to different backends
I have find it. at
- pytorch/torch/csrc/autograd/generated/python_torch_functionsEverything.cpp
- pytorch/torch/csrc/autograd/generated/VariableTypeEverything.cpp
- pytorch/torch/csrc/autograd/generated/Functions.cpp
- Finally, aten/src/ATen/native/Convolution.cpp