Skip to content

Commit 97c4ea4

Browse files
author
wayuanho
committed
fix bug
1 parent cdabebf commit 97c4ea4

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tf2onnx/tfonnx.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2299,21 +2299,21 @@ def rewrite_conv2d_with_pad(g, ops):
22992299
paddings = pad.inputs[1]
23002300

23012301
if not paddings.is_const():
2302-
return ops
2302+
continue
23032303
mode = pad.get_attr("mode")
23042304
if mode:
23052305
mode = mode.s.decode("utf-8").lower()
23062306
if mode not in [None, "constant"] or len(pad.input) >= 3:
2307-
return ops
2307+
continue
23082308
# Conv2D already has a pad
23092309
if conv.get_attr("padding") == "SAME":
2310-
return ops
2310+
continue
23112311

23122312
log.debug("merge pad [%s] into conv [%s]", pad.name, conv.name)
23132313
paddings_val = np.array(paddings.get_tensor_value())
23142314
# can't pad on batch or channel dimensions
23152315
if np.any(paddings_val[0]) or np.any(paddings_val[3]):
2316-
return ops
2316+
continue
23172317
paddings_val = paddings_val[1:3]
23182318
paddings_val = paddings_val.transpose().flatten()
23192319
g.replace_input(conv, conv.input[0], pad.input[0])

0 commit comments

Comments
 (0)