xutianming opened a new pull request #6240: URL: https://github.com/apache/incubator-tvm/pull/6240
I was compiling a TensorFlow saved model exported by TensorFlow 2.x. I encountered `ImportError` from `_get_tag_set` of `tensorflow_parser.py`. For TensorFlow 2.x, `get_saved_model_tag_sets` has been moved into `tensorflow.python.tools.saved_model_utils`. `_get_output_names` used `tf.Session()` to load saved model. By default, `tf.Session` exhausts all the memory on the device. Adding `config.gpu_options.allow_growth` prevents this behavior. This pull request refined above tiny issues. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
