38 lines
1.7 KiB
C++
38 lines
1.7 KiB
C++
/*!
|
|
**************************************************************************************************
|
|
* InternImage
|
|
* Copyright (c) 2022 OpenGVLab
|
|
* Licensed under The MIT License [see LICENSE for details]
|
|
**************************************************************************************************
|
|
* Modified from
|
|
*https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/tree/pytorch_1.0.0
|
|
**************************************************************************************************
|
|
*/
|
|
|
|
#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");
|
|
}
|