Yeah, well. The question was about cudnn_affine_grid_generator.
Without CuDNN, by applying the same method to search, it turns out that /aten/src/ATen/native/AffineGridGenerator.cpp 21 is the code that is used, independent of CPU vs. CUDA.