roi_align_rotated_v2

momo609 · Nov 27, 2024 · c5d347d · c5d347d
1 parent 5e0db76
commit c5d347d
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/mmcv/ops/csrc/pytorch/roi_align_rotated_v2.cpp b/mmcv/ops/csrc/pytorch/roi_align_rotated_v2.cpp
@@ -2,22 +2,22 @@
 #include "pytorch_cpp_helper.hpp"
 #include "pytorch_device_registry.hpp"
 
-void roi_align_rotated_v2_forward_impl(Tensor input, Tensor rois, Tensor output,
+void roi_align_rotated_v2_forward_impl(Tensor x, Tensor rois, Tensor y,
+                                    int pooled_h, int pooled_w,
                                     double spatial_scale, int sampling_ratio,
-                                    int pooled_height, int pooled_width,
                                     bool aligned, bool clockwise) {
-  DISPATCH_DEVICE_IMPL(roi_align_rotated_v2_forward_impl, input, rois, output,
-                       spatial_scale, sampling_ratio, pooled_height, pooled_width,
+  DISPATCH_DEVICE_IMPL(roi_align_rotated_v2_forward_impl, x, rois, y,
+                       pooled_h, pooled_w, spatial_scale, sampling_ratio,
                        aligned, clockwise);
 }
 
 
-void roi_align_rotated_v2_forward(Tensor input, Tensor rois, Tensor output,
+void roi_align_rotated_v2_forward(Tensor x, Tensor rois, Tensor y,
+                               int pooled_h, int pooled_w,
                                double spatial_scale, int sampling_ratio,
-                               int pooled_height, int pooled_width,
                                bool aligned, bool clockwise) {
-  roi_align_rotated_v2_forward_impl(input, rois, output, spatial_scale, sampling_ratio,
-                                    pooled_height, pooled_width, aligned, clockwise);
+  roi_align_rotated_v2_forward_impl(x, rois, y, pooled_h, pooled_w, 
+                                    spatial_scale, sampling_ratio, aligned, clockwise);
 }