Spaces:
Runtime error
Runtime error
/*! | |
************************************************************************************************** | |
* Deformable DETR | |
* Copyright (c) 2020 SenseTime. All Rights Reserved. | |
* Licensed under the Apache License, Version 2.0 [see LICENSE for details] | |
************************************************************************************************** | |
* Modified from https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/tree/pytorch_1.0.0 | |
************************************************************************************************** | |
*/ | |
at::Tensor ms_deform_attn_cuda_forward( | |
const at::Tensor &value, | |
const at::Tensor &spatial_shapes, | |
const at::Tensor &level_start_index, | |
const at::Tensor &sampling_loc, | |
const at::Tensor &attn_weight, | |
const int im2col_step); | |
std::vector<at::Tensor> ms_deform_attn_cuda_backward( | |
const at::Tensor &value, | |
const at::Tensor &spatial_shapes, | |
const at::Tensor &level_start_index, | |
const at::Tensor &sampling_loc, | |
const at::Tensor &attn_weight, | |
const at::Tensor &grad_output, | |
const int im2col_step); | |