Fix header includes for cpu (#1644)

gslotman · fmassa · commit 598b61d93357 · 2019-12-06T16:43:30.000+01:00
diff --git a/torchvision/csrc/cpu/ROIAlign_cpu.cpp b/torchvision/csrc/cpu/ROIAlign_cpu.cpp
@@ -1,5 +1,5 @@
 #include <ATen/TensorUtils.h>
-#include "cpu/vision_cpu.h"
+#include "vision_cpu.h"
 
 // implementation taken from Caffe2
 template <typename T>
diff --git a/torchvision/csrc/cpu/nms_cpu.cpp b/torchvision/csrc/cpu/nms_cpu.cpp
@@ -1,4 +1,4 @@
-#include "cpu/vision_cpu.h"
+#include "vision_cpu.h"
 
 template <typename scalar_t>
 at::Tensor nms_cpu_kernel(
