diff options
| author |  Katherine Wu <kathywu@google.com> | 2018-08-31 12:07:52 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-31 12:12:17 -0700 |
| commit | cda5ea80b86909fd20ff8a0f5ba914c5c03b876f (patch) | |
| tree | 9bfaa9eea182a31df1a7210eeae7f066d491483d /tensorflow/python/training | |
| parent | e894ca7c736c58a8e4c71f0c3f1b1f0c327fa924 (diff) | |
Roll forward of commit 069f808e5c0462819bcd6c73c75491b00cdd42c2 (rolling back rollback cl/210656847).
Fixing reference to _get_export_outputs_for_spec in TFMA (This function was refactored out, so the string has been removed from the list of methods that are copied from core Estimator).
*** Original change description ***
Automated rollback of commit 069f808e5c0462819bcd6c73c75491b00cdd42c2
PiperOrigin-RevId: 211122893
Diffstat (limited to 'tensorflow/python/training')
| -rw-r--r-- | tensorflow/python/training/checkpointable/util.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/python/training/checkpointable/util.py b/tensorflow/python/training/checkpointable/util.py index 45d217e8b1..13dddd37ac 100644 --- a/tensorflow/python/training/checkpointable/util.py +++ b/tensorflow/python/training/checkpointable/util.py @@ -685,6 +685,11 @@ def _serialize_object_graph(root_checkpointable, saveables_cache): saveables_cache=saveables_cache) +def named_saveables(root_checkpointable): + """Gather list of all SaveableObjects in the Checkpointable object.""" + return _serialize_object_graph(root_checkpointable, None)[0] + + def list_objects(root_checkpointable): """Traverse the object graph and list all accessible objects. |