| 34 | |
| 35 | |
| 36 | std::vector<torch::Tensor> corr_backward( |
| 37 | torch::Tensor fmap1, |
| 38 | torch::Tensor fmap2, |
| 39 | torch::Tensor coords, |
| 40 | torch::Tensor corr_grad, |
| 41 | int radius) { |
| 42 | CHECK_INPUT(fmap1); |
| 43 | CHECK_INPUT(fmap2); |
| 44 | CHECK_INPUT(coords); |
| 45 | CHECK_INPUT(corr_grad); |
| 46 | |
| 47 | return corr_cuda_backward(fmap1, fmap2, coords, corr_grad, radius); |
| 48 | } |
| 49 | |
| 50 | |
| 51 | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { |
nothing calls this directly
no outgoing calls
no test coverage detected