Unverified Commit ce342580 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Removed all backward methods from header files. (#3143)

parent 3c33f367
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
namespace vision { namespace vision {
namespace ops { namespace ops {
// C++ Forward
VISION_API at::Tensor deform_conv2d( VISION_API at::Tensor deform_conv2d(
const at::Tensor& input, const at::Tensor& input,
const at::Tensor& weight, const at::Tensor& weight,
...@@ -23,25 +22,5 @@ VISION_API at::Tensor deform_conv2d( ...@@ -23,25 +22,5 @@ VISION_API at::Tensor deform_conv2d(
int64_t offset_groups, int64_t offset_groups,
bool use_mask); 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 ops
} // namespace vision } // namespace vision
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
namespace vision { namespace vision {
namespace ops { namespace ops {
// C++ Forward
VISION_API at::Tensor nms( VISION_API at::Tensor nms(
const at::Tensor& dets, const at::Tensor& dets,
const at::Tensor& scores, const at::Tensor& scores,
......
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
namespace vision { namespace vision {
namespace ops { namespace ops {
// C++ Forward
VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align( VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align(
const at::Tensor& input, const at::Tensor& input,
const at::Tensor& rois, const at::Tensor& rois,
...@@ -15,19 +14,5 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align( ...@@ -15,19 +14,5 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align(
int64_t pooled_width, int64_t pooled_width,
int64_t sampling_ratio); 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 ops
} // namespace vision } // namespace vision
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
namespace vision { namespace vision {
namespace ops { namespace ops {
// C++ Forward
VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_pool( VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_pool(
const at::Tensor& input, const at::Tensor& input,
const at::Tensor& rois, const at::Tensor& rois,
...@@ -14,18 +13,5 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_pool( ...@@ -14,18 +13,5 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_pool(
int64_t pooled_height, int64_t pooled_height,
int64_t pooled_width); 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 ops
} // namespace vision } // namespace vision
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
namespace vision { namespace vision {
namespace ops { namespace ops {
// C++ Forward
VISION_API at::Tensor roi_align( VISION_API at::Tensor roi_align(
const at::Tensor& input, const at::Tensor& input,
const at::Tensor& rois, const at::Tensor& rois,
...@@ -16,19 +15,5 @@ VISION_API at::Tensor roi_align( ...@@ -16,19 +15,5 @@ VISION_API at::Tensor roi_align(
int64_t sampling_ratio, int64_t sampling_ratio,
bool aligned); 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 ops
} // namespace vision } // namespace vision
...@@ -6,7 +6,6 @@ ...@@ -6,7 +6,6 @@
namespace vision { namespace vision {
namespace ops { namespace ops {
// C++ Forward
VISION_API std::tuple<at::Tensor, at::Tensor> roi_pool( VISION_API std::tuple<at::Tensor, at::Tensor> roi_pool(
const at::Tensor& input, const at::Tensor& input,
const at::Tensor& rois, const at::Tensor& rois,
...@@ -14,18 +13,5 @@ VISION_API std::tuple<at::Tensor, at::Tensor> roi_pool( ...@@ -14,18 +13,5 @@ VISION_API std::tuple<at::Tensor, at::Tensor> roi_pool(
int64_t pooled_height, int64_t pooled_height,
int64_t pooled_width); 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 ops
} // namespace vision } // namespace vision
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment