Skip to content

Commit

Permalink
Removed all backward methods from header files. (#3143)
Browse files Browse the repository at this point in the history
  • Loading branch information
datumbox authored Dec 9, 2020
1 parent 3c33f36 commit ce34258
Show file tree
Hide file tree
Showing 6 changed files with 0 additions and 80 deletions.
21 changes: 0 additions & 21 deletions torchvision/csrc/deform_conv2d.h
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
namespace vision {
namespace ops {

// C++ Forward
VISION_API at::Tensor deform_conv2d(
const at::Tensor& input,
const at::Tensor& weight,
Expand All @@ -23,25 +22,5 @@ VISION_API at::Tensor deform_conv2d(
int64_t offset_groups,
bool use_mask);

// C++ Backward
VISION_API
std::tuple<at::Tensor, at::Tensor, at::Tensor, at::Tensor, at::Tensor>
_deform_conv2d_backward(
const at::Tensor& grad,
const at::Tensor& input,
const at::Tensor& weight,
const at::Tensor& offset,
const at::Tensor& mask,
const at::Tensor& bias,
int64_t stride_h,
int64_t stride_w,
int64_t pad_h,
int64_t pad_w,
int64_t dilation_h,
int64_t dilation_w,
int64_t groups,
int64_t offset_groups,
bool use_mask);

} // namespace ops
} // namespace vision
1 change: 0 additions & 1 deletion torchvision/csrc/nms.h
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
namespace vision {
namespace ops {

// C++ Forward
VISION_API at::Tensor nms(
const at::Tensor& dets,
const at::Tensor& scores,
Expand Down
15 changes: 0 additions & 15 deletions torchvision/csrc/ps_roi_align.h
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
namespace vision {
namespace ops {

// C++ Forward
VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align(
const at::Tensor& input,
const at::Tensor& rois,
Expand All @@ -15,19 +14,5 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align(
int64_t pooled_width,
int64_t sampling_ratio);

// C++ Backward
VISION_API at::Tensor _ps_roi_align_backward(
const at::Tensor& grad,
const at::Tensor& rois,
const at::Tensor& channel_mapping,
double spatial_scale,
int64_t pooled_height,
int64_t pooled_width,
int64_t sampling_ratio,
int64_t batch_size,
int64_t channels,
int64_t height,
int64_t width);

} // namespace ops
} // namespace vision
14 changes: 0 additions & 14 deletions torchvision/csrc/ps_roi_pool.h
Original file line number Diff line number Diff line change
Expand Up @@ -6,26 +6,12 @@
namespace vision {
namespace ops {

// C++ Forward
VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_pool(
const at::Tensor& input,
const at::Tensor& rois,
double spatial_scale,
int64_t pooled_height,
int64_t pooled_width);

// C++ Backward
VISION_API at::Tensor _ps_roi_pool_backward(
const at::Tensor& grad,
const at::Tensor& rois,
const at::Tensor& channel_mapping,
double spatial_scale,
int64_t pooled_height,
int64_t pooled_width,
int64_t batch_size,
int64_t channels,
int64_t height,
int64_t width);

} // namespace ops
} // namespace vision
15 changes: 0 additions & 15 deletions torchvision/csrc/roi_align.h
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
namespace vision {
namespace ops {

// C++ Forward
VISION_API at::Tensor roi_align(
const at::Tensor& input,
const at::Tensor& rois,
Expand All @@ -16,19 +15,5 @@ VISION_API at::Tensor roi_align(
int64_t sampling_ratio,
bool aligned);

// C++ Backward
VISION_API at::Tensor _roi_align_backward(
const at::Tensor& grad,
const at::Tensor& rois,
double spatial_scale,
int64_t pooled_height,
int64_t pooled_width,
int64_t batch_size,
int64_t channels,
int64_t height,
int64_t width,
int64_t sampling_ratio,
bool aligned);

} // namespace ops
} // namespace vision
14 changes: 0 additions & 14 deletions torchvision/csrc/roi_pool.h
Original file line number Diff line number Diff line change
Expand Up @@ -6,26 +6,12 @@
namespace vision {
namespace ops {

// C++ Forward
VISION_API std::tuple<at::Tensor, at::Tensor> roi_pool(
const at::Tensor& input,
const at::Tensor& rois,
double spatial_scale,
int64_t pooled_height,
int64_t pooled_width);

// C++ Backward
VISION_API at::Tensor _roi_pool_backward(
const at::Tensor& grad,
const at::Tensor& rois,
const at::Tensor& argmax,
double spatial_scale,
int64_t pooled_height,
int64_t pooled_width,
int64_t batch_size,
int64_t channels,
int64_t height,
int64_t width);

} // namespace ops
} // namespace vision

0 comments on commit ce34258

Please sign in to comment.