From d2ac97dfe4c9080694d0da07c4a4da80e0a3d3ae Mon Sep 17 00:00:00 2001 From: Yong Wu Date: Thu, 3 Oct 2019 14:40:34 -0700 Subject: [PATCH] [Bugfix][TF] reset graph after getting tag of savedmodel @zhiics @icemelon9 --- python/tvm/relay/frontend/tensorflow_parser.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/tvm/relay/frontend/tensorflow_parser.py b/python/tvm/relay/frontend/tensorflow_parser.py index 8105ef03aeca..d1c0bfe6a847 100644 --- a/python/tvm/relay/frontend/tensorflow_parser.py +++ b/python/tvm/relay/frontend/tensorflow_parser.py @@ -79,15 +79,16 @@ def _get_output_names(self): "InputConfiguration: Unable to import tensorflow which is " "required to restore from saved model.") tags = self._get_tag_set() + output_names = set() with tf.Session() as sess: meta_graph_def = tf.saved_model.loader.load(sess, tags, self._model_dir) - output_names = set() for sig_def in meta_graph_def.signature_def.values(): for output_tensor in sig_def.outputs.values(): output_names.add(output_tensor.name.replace(":0", "")) - return ",".join(output_names) + tf.reset_default_graph() + return ",".join(output_names) def _load_saved_model(self): """Load the tensorflow saved model."""