Fix header includes for cpu (#1644)

598b61d9 · gslotman · Francisco Massa · 3bd7a405 · 598b61d9 · 598b61d9
Commit 598b61d9 authored Dec 06, 2019 by gslotman Committed by Francisco Massa Dec 06, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchvision/csrc/cpu/ROIAlign_cpu.cpp torchvision/csrc/cpu/ROIAlign_cpu.cpp +1 -1

torchvision/csrc/cpu/nms_cpu.cpp torchvision/csrc/cpu/nms_cpu.cpp +1 -1

No files found.
--- a/torchvision/csrc/cpu/ROIAlign_cpu.cpp
+++ b/torchvision/csrc/cpu/ROIAlign_cpu.cpp
 #include <ATen/TensorUtils.h>
-#include "cpu/vision_cpu.h"
+#include "vision_cpu.h"

 // implementation taken from Caffe2
 template <typename T>

--- a/torchvision/csrc/cpu/nms_cpu.cpp
+++ b/torchvision/csrc/cpu/nms_cpu.cpp
-#include "cpu/vision_cpu.h"
+#include "vision_cpu.h"

 template <typename scalar_t>
 at::Tensor nms_cpu_kernel(