GroundingDINO_SOTA_Zero_Shot_Model
/
groundingdino
/models
/GroundingDINO
/csrc
/MsDeformAttn
/ms_deform_attn_cpu.h
/*! | |
************************************************************************************************** | |
* 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 | |
************************************************************************************************** | |
*/ | |
namespace groundingdino { | |
at::Tensor | |
ms_deform_attn_cpu_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_cpu_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); | |
} // namespace groundingdino | |