Spaces:
Running
on
T4
Running
on
T4
Grounded-Segment-Anything
/
transformers_4_35_0
/kernels
/deformable_detr
/cuda
/ms_deform_attn_cuda.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 | |
************************************************************************************************** | |
*/ | |
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); | |