diff options
author | Katherine Wu <kathywu@google.com> | 2018-08-28 19:00:00 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-28 19:04:24 -0700 |
commit | 069f808e5c0462819bcd6c73c75491b00cdd42c2 (patch) | |
tree | ce17bc2f4ec07ccd1baa790a335c192e0ab0fe2b /tensorflow/python/training | |
parent | 4a83b950aa2b5be238bed118acb006a3fe1c806e (diff) |
Export Keras model to SavedModel.
PiperOrigin-RevId: 210648154
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. |