diff options
author | 2016-03-02 08:25:38 -0800 | |
---|---|---|
committer | 2016-03-02 10:17:01 -0800 | |
commit | 9d8ea873db7246656c21f5860b53e34f14dd2841 (patch) | |
tree | c4f17a5d5b96356c29af0a7bf497d7773452b34b /tensorflow/contrib/util | |
parent | 2c40274367bfdcd226bc73d4515045ece2bf5de5 (diff) |
Expose tf.contrib.util.stripped_op_list_for_graph
C++ and Python use two different op registries, and in rare cases they can
actually be different. Thus, we need both functions available.
Also fix both Python and C++ to handle arbitrarily nested functions.
Change: 116142089
Diffstat (limited to 'tensorflow/contrib/util')
-rw-r--r-- | tensorflow/contrib/util/__init__.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/contrib/util/__init__.py b/tensorflow/contrib/util/__init__.py index d9ab533365..cb16610060 100644 --- a/tensorflow/contrib/util/__init__.py +++ b/tensorflow/contrib/util/__init__.py @@ -20,6 +20,7 @@ @@constant_value @@make_tensor_proto @@make_ndarray +@@stripped_op_list_for_graph """ @@ -31,6 +32,8 @@ 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 stripped_op_list_for_graph # TODO(irving): Use make_all here. -__all__ = ['constant_value', 'make_tensor_proto', 'make_ndarray'] +__all__ = ['constant_value', 'make_tensor_proto', 'make_ndarray', + 'stripped_op_list_for_graph'] |