diff options
author | Yifei Feng <fengyifei2026@gmail.com> | 2017-10-16 10:45:33 -0700 |
---|---|---|
committer | Vijay Vasudevan <vrv@google.com> | 2017-10-16 10:45:33 -0700 |
commit | 2f916fcf2f49f8a30d8460c1c8d7812d637ab0e4 (patch) | |
tree | a7236c2624a5e688437f14c5b8f0bb929a0b6022 | |
parent | 2ebbce8e82d8d07bee6b8be14a3961ebdef977a0 (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.py | 1 |
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 |