Merge branch 'main' into food-101-dataset

Author: pmeier

torchvision/csrc/io/image/image.cpp

@@ -1,5 +1,6 @@
 #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

@@ -1,6 +1,7 @@
 #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

@@ -1,6 +1,7 @@
 #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

@@ -1,6 +1,7 @@
 #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

@@ -1,6 +1,7 @@
 #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

@@ -1,6 +1,7 @@
 #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

@@ -1,6 +1,7 @@
 #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

@@ -1,5 +1,7 @@
 #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

@@ -1,5 +1,7 @@
 #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

@@ -1,5 +1,7 @@
 #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

@@ -1,5 +1,7 @@
 #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

@@ -1,5 +1,7 @@
 #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

@@ -1,5 +1,7 @@
 #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

@@ -1,5 +1,7 @@
 #include "roi_pool.h"
 
+#include <ATen/core/dispatch/Dispatcher.h>
+#include <torch/library.h>
 #include <torch/types.h>
 
 namespace vision {