Fix missing Torch includes (#5118)

* Fix missing Torch includes

* Import op_registration directly

torchvision/csrc/io/image/image.cpp

 #include "image.h"
 
+#include <ATen/core/op_registration/op_registration.h>
 #include <Python.h>
 
 // If we are in a Windows environment, we need to define

torchvision/csrc/ops/autocast/deform_conv2d_kernel.cpp

 #include "../deform_conv2d.h"
 
 #include <ATen/autocast_mode.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/autocast/nms_kernel.cpp

 #include "../nms.h"
 
 #include <ATen/autocast_mode.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/autocast/ps_roi_align_kernel.cpp

 #include "../ps_roi_align.h"
 
 #include <ATen/autocast_mode.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/autocast/ps_roi_pool_kernel.cpp

 #include "../ps_roi_pool.h"
 
 #include <ATen/autocast_mode.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/autocast/roi_align_kernel.cpp

 #include "../roi_align.h"
 
 #include <ATen/autocast_mode.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/autocast/roi_pool_kernel.cpp

 #include "../roi_pool.h"
 
 #include <ATen/autocast_mode.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/deform_conv2d.cpp

 #include "deform_conv2d.h"
 
+#include <ATen/core/dispatch/Dispatcher.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/interpolate_aa.cpp

 #include "interpolate_aa.h"
 
+#include <ATen/core/dispatch/Dispatcher.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/nms.cpp

 #include "nms.h"
 
+#include <ATen/core/dispatch/Dispatcher.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/ps_roi_align.cpp

 #include "ps_roi_align.h"
 
+#include <ATen/core/dispatch/Dispatcher.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/ps_roi_pool.cpp

 #include "ps_roi_pool.h"
 
+#include <ATen/core/dispatch/Dispatcher.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/roi_align.cpp

 #include "roi_align.h"
 
+#include <ATen/core/dispatch/Dispatcher.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {

torchvision/csrc/ops/roi_pool.cpp

 #include "roi_pool.h"
 
+#include <ATen/core/dispatch/Dispatcher.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {