| 50 | #endif |
| 51 | |
| 52 | int deform_conv_forward(at::Tensor input, at::Tensor weight, |
| 53 | at::Tensor offset, at::Tensor output, |
| 54 | at::Tensor columns, at::Tensor ones, int kW, |
| 55 | int kH, int dW, int dH, int padW, int padH, |
| 56 | int dilationW, int dilationH, int group, |
| 57 | int deformable_group, int im2col_step) { |
| 58 | if (input.device().is_cuda()) { |
| 59 | #ifdef WITH_CUDA |
| 60 | return deform_conv_forward_cuda(input, weight, offset, output, columns, |
| 61 | ones, kW, kH, dW, dH, padW, padH, dilationW, dilationH, group, |
| 62 | deformable_group, im2col_step); |
| 63 | #else |
| 64 | AT_ERROR("deform conv is not compiled with GPU support"); |
| 65 | #endif |
| 66 | } |
| 67 | AT_ERROR("deform conv is not implemented on CPU"); |
| 68 | } |
| 69 | |
| 70 | int deform_conv_backward_input(at::Tensor input, at::Tensor offset, |
| 71 | at::Tensor gradOutput, at::Tensor gradInput, |
nothing calls this directly
no test coverage detected