diff options
| author |  Martin Wicke <wicke@google.com> | 2016-09-21 13:16:48 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-21 14:31:57 -0700 |
| commit | 999b794c137d12d73adbf41dcbe9383a0cd94769 (patch) | |
| tree | 90a58d6f358e4dff917912d46cbb1fbf89536ff4 /tensorflow/contrib/quantization | |
| parent | 8e308a8493436fdd7b555837d36459fb232bc7c2 (diff) | |
Merge changes from github.
Change: 133874452
Diffstat (limited to 'tensorflow/contrib/quantization')
| -rw-r--r-- | tensorflow/contrib/quantization/tools/quantize_graph.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/quantization/tools/quantize_graph.py b/tensorflow/contrib/quantization/tools/quantize_graph.py index 0a814dadae..5ded556691 100644 --- a/tensorflow/contrib/quantization/tools/quantize_graph.py +++ b/tensorflow/contrib/quantization/tools/quantize_graph.py @@ -981,7 +981,7 @@ def main(unused_args): return -1 tf_graph = tf.GraphDef() - with tf.gfile.Open(FLAGS.input, "r") as f: + with tf.gfile.Open(FLAGS.input, "rb") as f: data = f.read() tf_graph.ParseFromString(data) @@ -993,7 +993,7 @@ def main(unused_args): output_graph = rewriter.rewrite(FLAGS.output_node_names.split(",")) - f = tf.gfile.FastGFile(FLAGS.output, "w") + f = tf.gfile.FastGFile(FLAGS.output, "wb") f.write(output_graph.SerializeToString()) return 0 |