@@ -4,9 +4,9 @@ using namespace NPU_NAME_SPACE;
44using namespace std ;
55
66void roi_align_rotated_forward_npu (Tensor input, Tensor rois, Tensor output,
7- int aligned_height, int aligned_width,
8- float spatial_scale, int sampling_ratio,
9- bool aligned, bool clockwise) {
7+ int aligned_height, int aligned_width,
8+ float spatial_scale, int sampling_ratio,
9+ bool aligned, bool clockwise) {
1010 int64_t aligned_height_64 = aligned_height;
1111 int64_t aligned_width_64 = aligned_width;
1212 int64_t sampling_ratio_64 = sampling_ratio;
@@ -25,10 +25,10 @@ void roi_align_rotated_forward_npu(Tensor input, Tensor rois, Tensor output,
2525}
2626
2727void roi_align_rotated_backward_npu (Tensor top_grad, Tensor rois,
28- Tensor bottom_grad, int aligned_height,
29- int aligned_width, float spatial_scale,
30- int sampling_ratio, bool aligned,
31- bool clockwise) {
28+ Tensor bottom_grad, int aligned_height,
29+ int aligned_width, float spatial_scale,
30+ int sampling_ratio, bool aligned,
31+ bool clockwise) {
3232 int64_t aligned_height_64 = aligned_height;
3333 int64_t aligned_width_64 = aligned_width;
3434 int64_t sampling_ratio_64 = sampling_ratio;
@@ -60,5 +60,7 @@ void roi_align_rotated_backward_impl(Tensor top_grad, Tensor rois,
6060 int sampling_ratio, bool aligned,
6161 bool clockwise);
6262
63- REGISTER_NPU_IMPL (roi_align_rotated_forward_impl, roi_align_rotated_forward_npu);
64- REGISTER_NPU_IMPL (roi_align_rotated_backward_impl, roi_align_rotated_backward_npu);
63+ REGISTER_NPU_IMPL (roi_align_rotated_forward_impl,
64+ roi_align_rotated_forward_npu);
65+ REGISTER_NPU_IMPL (roi_align_rotated_backward_impl,
66+ roi_align_rotated_backward_npu);
0 commit comments