Replace np.int with np.int32 (#17484)

apache · Oct 22, 2024 · d973b33 · d973b33
1 parent 3219b49
commit d973b33
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/topi/test_topi_depthwise_conv2d_back_input.py b/tests/python/topi/test_topi_depthwise_conv2d_back_input.py
@@ -36,8 +36,8 @@ def verify_depthwise_conv2d_back_input(
     stride_w = stride_h
     padding_w = padding_h
 
-    out_h = np.int((in_h + 2 * padding_h - filter_h) / stride_h + 1)
-    out_w = np.int((in_w + 2 * padding_w - filter_w) / stride_w + 1)
+    out_h = np.int32((in_h + 2 * padding_h - filter_h) / stride_h + 1)
+    out_w = np.int32((in_w + 2 * padding_w - filter_w) / stride_w + 1)
     out_channel = in_channel * channel_multiplier
 
     ishape = [batch, in_h, in_w, in_channel]