From 2c2cf71ce484b4b8426738324db1b278b8dbe6cc Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9D=B4=EC=84=B1=EC=9E=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 5 Aug 2019 10:52:59 +0900 Subject: [PATCH] Update scheduled deprecated API. (#6163) * Change old API `gfile.FastGFile` to `tf.gfile.GFile` Signed-off-by: Sung-Jae Lee --- tools/pbfile_tool/pb_info.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tools/pbfile_tool/pb_info.py b/tools/pbfile_tool/pb_info.py index 110b15e..7add94f 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()) -- 2.7.4