Skip to content

Commit 4eab7a6

Browse files
authored
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.
1 parent 45d9a30 commit 4eab7a6

File tree

6 files changed

+12
-5
lines changed

6 files changed

+12
-5
lines changed

torchvision/csrc/ops/deform_conv2d.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ VISION_API at::Tensor deform_conv2d(
2424

2525
namespace detail {
2626

27-
VISION_API
2827
std::tuple<at::Tensor, at::Tensor, at::Tensor, at::Tensor, at::Tensor>
2928
_deform_conv2d_backward(
3029
const at::Tensor& grad,

torchvision/csrc/ops/ops.h

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
#pragma once
2+
3+
#include "deform_conv2d.h"
4+
#include "nms.h"
5+
#include "ps_roi_align.h"
6+
#include "ps_roi_pool.h"
7+
#include "roi_align.h"
8+
#include "roi_pool.h"

torchvision/csrc/ops/ps_roi_align.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align(
1616

1717
namespace detail {
1818

19-
VISION_API at::Tensor _ps_roi_align_backward(
19+
at::Tensor _ps_roi_align_backward(
2020
const at::Tensor& grad,
2121
const at::Tensor& rois,
2222
const at::Tensor& channel_mapping,

torchvision/csrc/ops/ps_roi_pool.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_pool(
1515

1616
namespace detail {
1717

18-
VISION_API at::Tensor _ps_roi_pool_backward(
18+
at::Tensor _ps_roi_pool_backward(
1919
const at::Tensor& grad,
2020
const at::Tensor& rois,
2121
const at::Tensor& channel_mapping,

torchvision/csrc/ops/roi_align.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ VISION_API at::Tensor roi_align(
1717

1818
namespace detail {
1919

20-
VISION_API at::Tensor _roi_align_backward(
20+
at::Tensor _roi_align_backward(
2121
const at::Tensor& grad,
2222
const at::Tensor& rois,
2323
double spatial_scale,

torchvision/csrc/ops/roi_pool.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> roi_pool(
1515

1616
namespace detail {
1717

18-
VISION_API at::Tensor _roi_pool_backward(
18+
at::Tensor _roi_pool_backward(
1919
const at::Tensor& grad,
2020
const at::Tensor& rois,
2121
const at::Tensor& argmax,

0 commit comments

Comments
 (0)