in tf, two branches of cond can have different shapes

zhijxu-MS · zhijxu-MS · commit 72ef2d373f94 · 2019-04-12T16:13:17.000+08:00
diff --git a/tf2onnx/rewriter/cond_rewriter.py b/tf2onnx/rewriter/cond_rewriter.py
@@ -127,7 +127,7 @@ def _get_output_shape_dtype(self, cond_context):
                         false_dtype
                     )
                 )
-            output_shapes.append(utils.merge_shapes(true_shape, false_shape))
+            output_shapes.append(utils.create_vague_shape_like(utils.merge_shapes(true_shape, false_shape)))
             output_dtypes.append(true_dtype)
         return output_shapes, output_dtypes
 

Original file line number	Diff line number	Diff line change
`@@ -127,7 +127,7 @@ def _get_output_shape_dtype(self, cond_context):`
`127`	`127`	`false_dtype`
`128`	`128`	`)`
`129`	`129`	`)`
`130`		`- output_shapes.append(utils.merge_shapes(true_shape, false_shape))`
	`130`	`+ output_shapes.append(utils.create_vague_shape_like(utils.merge_shapes(true_shape, false_shape)))`
`131`	`131`	`output_dtypes.append(true_dtype)`
`132`	`132`	`return output_shapes, output_dtypes`
`133`	`133`