| 105 | } |
| 106 | |
| 107 | void modulated_deform_conv_forward( |
| 108 | at::Tensor input, at::Tensor weight, at::Tensor bias, at::Tensor ones, |
| 109 | at::Tensor offset, at::Tensor mask, at::Tensor output, at::Tensor columns, |
| 110 | int kernel_h, int kernel_w, const int stride_h, const int stride_w, |
| 111 | const int pad_h, const int pad_w, const int dilation_h, |
| 112 | const int dilation_w, const int group, const int deformable_group, |
| 113 | const bool with_bias) { |
| 114 | if (input.device().is_cuda()) { |
| 115 | #ifdef WITH_CUDA |
| 116 | return modulated_deform_conv_cuda_forward(input, weight, bias, ones, |
| 117 | offset, mask, output, columns, kernel_h, kernel_w, stride_h, |
| 118 | stride_w, pad_h, pad_w, dilation_h, dilation_w, group, |
| 119 | deformable_group, with_bias); |
| 120 | #else |
| 121 | AT_ERROR("modulated deform conv is not compiled with GPU support"); |
| 122 | #endif |
| 123 | } |
| 124 | AT_ERROR("modulated deform conv is not implemented on CPU"); |
| 125 | } |
| 126 | |
| 127 | void modulated_deform_conv_backward( |
| 128 | at::Tensor input, at::Tensor weight, at::Tensor bias, at::Tensor ones, |
nothing calls this directly
no test coverage detected