[BugFix][Topi] Fix 'duplicated iterator names in the compute definiti…

…on' bug of roi_align (#11322)
apache · May 16, 2022 · 02d57bb · 02d57bb
1 parent eed21ee
commit 02d57bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tvm/topi/vision/rcnn/roi_align.py b/python/tvm/topi/vision/rcnn/roi_align.py
@@ -58,8 +58,8 @@ def _sample_common(
         roi_bin_grid_w = te.ceil(roi_w / pooled_size_w).astype("int32")
 
     count = roi_bin_grid_h * roi_bin_grid_w
-    rh = te.reduce_axis((0, roi_bin_grid_h))
-    rw = te.reduce_axis((0, roi_bin_grid_w))
+    rh = te.reduce_axis((0, roi_bin_grid_h), name="rh")
+    rw = te.reduce_axis((0, roi_bin_grid_w), name="rw")
     roi_start_h += ph * bin_h
     roi_start_w += pw * bin_w