aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Manjunath Kudlur <keveman@gmail.com>2016-01-18 14:00:05 -0800
committerGravatar Manjunath Kudlur <keveman@gmail.com>2016-01-18 14:00:58 -0800
commit8a445e54b893f313ca44c7ef124b1884f4994e95 (patch)
tree24885ed811f09e263883c0cac4da2ba0871cabf6
parent04bad79a588a09f7642544c5e1cc168f7dfbcab4 (diff)
DEFINE_bool -> DEFINE_boolean
Change: 112425239
-rw-r--r--tensorflow/python/tools/freeze_graph.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/python/tools/freeze_graph.py b/tensorflow/python/tools/freeze_graph.py
index cb291b34cb..3abb55eaed 100644
--- a/tensorflow/python/tools/freeze_graph.py
+++ b/tensorflow/python/tools/freeze_graph.py
@@ -58,16 +58,16 @@ tf.app.flags.DEFINE_string("input_checkpoint", "",
"""TensorFlow variables file to load.""")
tf.app.flags.DEFINE_string("output_graph", "",
"""Output 'GraphDef' file name.""")
-tf.app.flags.DEFINE_bool("input_binary", False,
- """Whether the input files are in binary format.""")
+tf.app.flags.DEFINE_boolean("input_binary", False,
+ """Whether the input files are in binary format.""")
tf.app.flags.DEFINE_string("output_node_names", "",
"""The name of the output nodes, comma separated.""")
tf.app.flags.DEFINE_string("restore_op_name", "save/restore_all",
"""The name of the master restore operator.""")
tf.app.flags.DEFINE_string("filename_tensor_name", "save/Const:0",
"""The name of the tensor holding the save path.""")
-tf.app.flags.DEFINE_bool("clear_devices", True,
- """Whether to remove device specifications.""")
+tf.app.flags.DEFINE_boolean("clear_devices", True,
+ """Whether to remove device specifications.""")
def set_attr_dtype(node, key, value):