Please, where can I find the implementation of torch._C._nn._upsample_nearest_exact3d
The dispatching should be done here which should then call into the backend. E.g. the CPU implementation can be found here and you could use the kernel names to check for other backends.