|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#include <vector> |
|
|
|
#include <ATen/ATen.h> |
|
#include <ATen/cuda/CUDAContext.h> |
|
|
|
at::Tensor dcnv3_cpu_forward(const at::Tensor &input, const at::Tensor &offset, |
|
const at::Tensor &mask, const int kernel_h, |
|
const int kernel_w, const int stride_h, |
|
const int stride_w, const int pad_h, |
|
const int pad_w, const int dilation_h, |
|
const int dilation_w, const int group, |
|
const int group_channels, const float offset_scale, |
|
const int im2col_step) { |
|
AT_ERROR("Not implement on cpu"); |
|
} |
|
|
|
std::vector<at::Tensor> |
|
dcnv3_cpu_backward(const at::Tensor &input, const at::Tensor &offset, |
|
const at::Tensor &mask, const int kernel_h, |
|
const int kernel_w, const int stride_h, const int stride_w, |
|
const int pad_h, const int pad_w, const int dilation_h, |
|
const int dilation_w, const int group, |
|
const int group_channels, const float offset_scale, |
|
const at::Tensor &grad_output, const int im2col_step) { |
|
AT_ERROR("Not implement on cpu"); |
|
} |
|
|