| 21 | #define CHECK_INPUT(x) CHECK_CUDA(x); CHECK_CONTIGUOUS(x) |
| 22 | |
| 23 | std::vector<torch::Tensor> corr_forward( |
| 24 | torch::Tensor fmap1, |
| 25 | torch::Tensor fmap2, |
| 26 | torch::Tensor coords, |
| 27 | int radius) { |
| 28 | CHECK_INPUT(fmap1); |
| 29 | CHECK_INPUT(fmap2); |
| 30 | CHECK_INPUT(coords); |
| 31 | |
| 32 | return corr_cuda_forward(fmap1, fmap2, coords, radius); |
| 33 | } |
| 34 | |
| 35 | |
| 36 | std::vector<torch::Tensor> corr_backward( |
nothing calls this directly
no outgoing calls
no test coverage detected