diff options
author | Josh Levenberg <josh11b@tensorflow.org> | 2016-04-29 09:28:55 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-04-29 10:32:12 -0700 |
commit | ec4d8840a5b42a2d88cef732b2c7f3e9f1e4b18d (patch) | |
tree | f2b5bc7236862d408078267000c6063125feebe7 /tensorflow/contrib/util | |
parent | 7e92390e7a39e66284e38a842190463e872c5c1b (diff) |
Add function ops_used_by_graph_def(), with the functionality of
stripped_op_list_for_graph() except it returns string names instead
of OpDefs.
Change: 121128648
Diffstat (limited to 'tensorflow/contrib/util')
-rw-r--r-- | tensorflow/contrib/util/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/util/__init__.py b/tensorflow/contrib/util/__init__.py index 66a168afcb..a8746db222 100644 --- a/tensorflow/contrib/util/__init__.py +++ b/tensorflow/contrib/util/__init__.py @@ -20,6 +20,7 @@ @@constant_value @@make_tensor_proto @@make_ndarray +@ops_used_by_graph_def @@stripped_op_list_for_graph """ @@ -32,6 +33,7 @@ from __future__ import print_function from tensorflow.python.framework.tensor_util import constant_value from tensorflow.python.framework.tensor_util import make_tensor_proto from tensorflow.python.framework.tensor_util import MakeNdarray as make_ndarray +from tensorflow.python.training.saver import ops_used_by_graph_def from tensorflow.python.training.saver import stripped_op_list_for_graph from tensorflow.python.util.all_util import make_all |