Removing VISION_API from backward() methods and adding an ops.h (#3163)

* Removing VISION_API from backward() methods and adding a ops.h * Fixing clang format.
pytorch · Dec 14, 2020 · 4eab7a6 · 4eab7a6
1 parent 45d9a30
commit 4eab7a6
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 5 deletions.
diff --git a/torchvision/csrc/ops/deform_conv2d.h b/torchvision/csrc/ops/deform_conv2d.h
@@ -24,7 +24,6 @@ VISION_API at::Tensor deform_conv2d(
 
 namespace detail {
 
-VISION_API
 std::tuple<at::Tensor, at::Tensor, at::Tensor, at::Tensor, at::Tensor>
 _deform_conv2d_backward(
     const at::Tensor& grad,

diff --git a/torchvision/csrc/ops/ops.h b/torchvision/csrc/ops/ops.h
@@ -0,0 +1,8 @@
+#pragma once
+
+#include "deform_conv2d.h"
+#include "nms.h"
+#include "ps_roi_align.h"
+#include "ps_roi_pool.h"
+#include "roi_align.h"
+#include "roi_pool.h"
diff --git a/torchvision/csrc/ops/ps_roi_align.h b/torchvision/csrc/ops/ps_roi_align.h
@@ -16,7 +16,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align(
 
 namespace detail {
 
-VISION_API at::Tensor _ps_roi_align_backward(
+at::Tensor _ps_roi_align_backward(
     const at::Tensor& grad,
     const at::Tensor& rois,
     const at::Tensor& channel_mapping,

diff --git a/torchvision/csrc/ops/ps_roi_pool.h b/torchvision/csrc/ops/ps_roi_pool.h
@@ -15,7 +15,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_pool(
 
 namespace detail {
 
-VISION_API at::Tensor _ps_roi_pool_backward(
+at::Tensor _ps_roi_pool_backward(
     const at::Tensor& grad,
     const at::Tensor& rois,
     const at::Tensor& channel_mapping,

diff --git a/torchvision/csrc/ops/roi_align.h b/torchvision/csrc/ops/roi_align.h
@@ -17,7 +17,7 @@ VISION_API at::Tensor roi_align(
 
 namespace detail {
 
-VISION_API at::Tensor _roi_align_backward(
+at::Tensor _roi_align_backward(
     const at::Tensor& grad,
     const at::Tensor& rois,
     double spatial_scale,

diff --git a/torchvision/csrc/ops/roi_pool.h b/torchvision/csrc/ops/roi_pool.h
@@ -15,7 +15,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> roi_pool(
 
 namespace detail {
 
-VISION_API at::Tensor _roi_pool_backward(
+at::Tensor _roi_pool_backward(
     const at::Tensor& grad,
     const at::Tensor& rois,
     const at::Tensor& argmax,