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

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.
parent 45d9a304
...@@ -24,7 +24,6 @@ VISION_API at::Tensor deform_conv2d( ...@@ -24,7 +24,6 @@ VISION_API at::Tensor deform_conv2d(
namespace detail { namespace detail {
VISION_API
std::tuple<at::Tensor, at::Tensor, at::Tensor, at::Tensor, at::Tensor> std::tuple<at::Tensor, at::Tensor, at::Tensor, at::Tensor, at::Tensor>
_deform_conv2d_backward( _deform_conv2d_backward(
const at::Tensor& grad, const at::Tensor& grad,
......
#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"
...@@ -16,7 +16,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align( ...@@ -16,7 +16,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_align(
namespace detail { namespace detail {
VISION_API at::Tensor _ps_roi_align_backward( at::Tensor _ps_roi_align_backward(
const at::Tensor& grad, const at::Tensor& grad,
const at::Tensor& rois, const at::Tensor& rois,
const at::Tensor& channel_mapping, const at::Tensor& channel_mapping,
......
...@@ -15,7 +15,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_pool( ...@@ -15,7 +15,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> ps_roi_pool(
namespace detail { namespace detail {
VISION_API at::Tensor _ps_roi_pool_backward( at::Tensor _ps_roi_pool_backward(
const at::Tensor& grad, const at::Tensor& grad,
const at::Tensor& rois, const at::Tensor& rois,
const at::Tensor& channel_mapping, const at::Tensor& channel_mapping,
......
...@@ -17,7 +17,7 @@ VISION_API at::Tensor roi_align( ...@@ -17,7 +17,7 @@ VISION_API at::Tensor roi_align(
namespace detail { namespace detail {
VISION_API at::Tensor _roi_align_backward( at::Tensor _roi_align_backward(
const at::Tensor& grad, const at::Tensor& grad,
const at::Tensor& rois, const at::Tensor& rois,
double spatial_scale, double spatial_scale,
......
...@@ -15,7 +15,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> roi_pool( ...@@ -15,7 +15,7 @@ VISION_API std::tuple<at::Tensor, at::Tensor> roi_pool(
namespace detail { namespace detail {
VISION_API at::Tensor _roi_pool_backward( at::Tensor _roi_pool_backward(
const at::Tensor& grad, const at::Tensor& grad,
const at::Tensor& rois, const at::Tensor& rois,
const at::Tensor& argmax, const at::Tensor& argmax,
......
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