diff options
| author | 이성재/On-Device Lab(SR)/Principal Engineer/삼성전자 <sj925.lee@samsung.com> | 2019-08-05 10:52:59 +0900 |
|---|---|---|
| committer | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2019-08-05 10:52:59 +0900 |
| commit | 2c2cf71ce484b4b8426738324db1b278b8dbe6cc (patch) | |
| tree | 3d577c84377d46fd8f94c9aa5c66296d89364278 /tools | |
| parent | a69c74ad8bf66f5a302cfd84f23876706f7065d8 (diff) | |
| download | nnfw-2c2cf71ce484b4b8426738324db1b278b8dbe6cc.tar.gz nnfw-2c2cf71ce484b4b8426738324db1b278b8dbe6cc.tar.bz2 nnfw-2c2cf71ce484b4b8426738324db1b278b8dbe6cc.zip | |
Update scheduled deprecated API. (#6163)
* Change old API `gfile.FastGFile` to `tf.gfile.GFile`
Signed-off-by: Sung-Jae Lee <sj925.lee@samsung.com>
Diffstat (limited to 'tools')
| -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()) |