Skip to content

Commit 71747f3

Browse files
committed
fix bug: reshape op only support with one dim is -1
1 parent 42ade07 commit 71747f3

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,9 @@ def post_optimize_action(self):
8181
input_shape = self._g.get_shape(op.input[0])
8282
if not input_shape:
8383
continue
84+
# reshape only supports one dime is -1
85+
if input_shape.count(-1) > 1:
86+
continue
8487

8588
new_shape = []
8689
# when transpose is NHWC_TO_NCHW

tf2onnx/tfonnx.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -473,6 +473,7 @@ def rewrite_incomplete_type_support_rs5(g, ops):
473473
def rewrite_incomplete_type_support_rs6(g, ops):
474474
return rewrite_incomplete_type_support(g, ops, [
475475
"Div",
476+
"Greater",
476477
"IsNaN",
477478
"ReduceSum",
478479
"Slice",

0 commit comments

Comments
 (0)