keep condition

titaiwangms · titaiwangms · commit 505a942bbb88 · 2025-01-22T23:43:46.000Z
diff --git a/onnx/reference/ops/op_pool_common.py b/onnx/reference/ops/op_pool_common.py
@@ -229,7 +229,7 @@ def pool(
                     strides[i] * shape[i + 2] + (1 + (kernel[i] - 1) * dilations[i]),
                     dilations[i],
                 )
-                # if num < x_shape[i + 2] + pads[i] + pads[i + spatial_size]
+                if num < x_shape[i + 2] + pads[i] + pads[i + spatial_size]
             )
             window_vals = np.array(
                 [window[indices] for indices in itertools.product(elements)]

Original file line number	Diff line number	Diff line change
`@@ -229,7 +229,7 @@ def pool(`
`229`	`229`	`strides[i] * shape[i + 2] + (1 + (kernel[i] - 1) * dilations[i]),`
`230`	`230`	`dilations[i],`
`231`	`231`	`)`
`232`		`- # if num < x_shape[i + 2] + pads[i] + pads[i + spatial_size]`
	`232`	`+ if num < x_shape[i + 2] + pads[i] + pads[i + spatial_size]`
`233`	`233`	`)`
`234`	`234`	`window_vals = np.array(`
`235`	`235`	`[window[indices] for indices in itertools.product(elements)]`