reload tf graph when shape is updated

lucienwang1009 · lucienwang1009 · commit f5a7db464792 · 2019-05-20T09:20:45.000Z
diff --git a/tf2onnx/shape_inference.py b/tf2onnx/shape_inference.py
@@ -92,7 +92,8 @@ def infer_shape_for_graph(tf_graph):
             updated = infer_shape_for_op(o)
             if updated:
                 shape_updated = True
-        tf_graph = reload_tf_graph(tf_graph)
+        if shape_updated:
+            tf_graph = reload_tf_graph(tf_graph)
     return tf_graph
 
 
