diff options
Diffstat (limited to 'tools/pbfile_tool/pb_info.py')
-rwxr-xr-x | tools/pbfile_tool/pb_info.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tools/pbfile_tool/pb_info.py b/tools/pbfile_tool/pb_info.py index 110b15e02..7add94fa8 100755 --- a/tools/pbfile_tool/pb_info.py +++ b/tools/pbfile_tool/pb_info.py @@ -16,7 +16,6 @@ import tensorflow as tf from google.protobuf import text_format -from tensorflow.python.platform import gfile from tensorflow.python.tools import freeze_graph from tensorflow.python.tools import optimize_for_inference_lib @@ -40,7 +39,7 @@ def importGraphIntoSession(sess, filename): assert sess (_, _, ext) = splitDirFilenameExt(filename) if (ext.lower() == 'pb'): - with gfile.FastGFile(filename, 'rb') as f: + with tf.gfile.GFile(filename, 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) |