diff options
| author |  Geoffrey Irving <geoffreyi@google.com> | 2016-02-29 15:55:27 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-02-29 16:06:37 -0800 |
| commit | d4b17f62fa09a10869d0f5d70e5afde833b23f0f (patch) | |
| tree | 4af67f7a7833fabd4e5c1a2ddff55a771adf74a5 /tensorflow/contrib/util | |
| parent | de50fc8cc523b3403ba54a29174b05a771971b99 (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: 115915068
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'] |