aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Yifei Feng <fengyifei2026@gmail.com>2017-10-16 10:45:33 -0700
committerGravatar Vijay Vasudevan <vrv@google.com>2017-10-16 10:45:33 -0700
commit2f916fcf2f49f8a30d8460c1c8d7812d637ab0e4 (patch)
treea7236c2624a5e688437f14c5b8f0bb929a0b6022
parent2ebbce8e82d8d07bee6b8be14a3961ebdef977a0 (diff)
Add freeze_graph to CONSOLE_SCRIPTS. (#13739)
Make freeze_graph accessible from command line from pip package.
-rw-r--r--tensorflow/tools/pip_package/setup.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
index 2ffaf7b1aa..02723f3e79 100644
--- a/tensorflow/tools/pip_package/setup.py
+++ b/tensorflow/tools/pip_package/setup.py
@@ -67,6 +67,7 @@ if sys.version_info < (3, 4):
# pylint: disable=line-too-long
CONSOLE_SCRIPTS = [
+ 'freeze_graph = tensorflow.python.tools.freeze_graph:main',
'saved_model_cli = tensorflow.python.tools.saved_model_cli:main',
# We need to keep the TensorBoard command, even though the console script
# is now declared by the tensorboard pip package. If we remove the