diff options
author | 2018-05-09 15:56:43 -0700 | |
---|---|---|
committer | 2018-05-09 15:59:21 -0700 | |
commit | ef58a46b730155717f1b03abb20767c1924ad05e (patch) | |
tree | 6d9509f18b878d07f9e320a566a559c83bac2613 /tensorflow/python/BUILD | |
parent | 22b8b9a528c658144a16dce19ba506561abae2ee (diff) |
Support saving Python state with object-based checkpoints
Allows SaveableObjects to specify feed dict addition callbacks for object-based saving.
For now just saves get_config() with Layers. Doesn't do any loading, and there isn't quite enough information to reconstruct a Model yet (needs topology).
My plan is to get Models to the point where they can be reconstructed from object-based checkpoints (probably one more change), add in SavedModel export (assuming no dynamic control flow for now), then add this "SavedModel+Python" format to Model.save / load_model.
PiperOrigin-RevId: 196043183
Diffstat (limited to 'tensorflow/python/BUILD')
-rw-r--r-- | tensorflow/python/BUILD | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index f7cbaec6ab..8b904a16c7 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3036,9 +3036,12 @@ py_library( srcs_version = "PY2AND3", deps = [ ":array_ops", + ":constant_op", + ":control_flow_ops", ":dtypes", ":io_ops_gen", ":ops", + ":saveable_object", ":util", "//tensorflow/python/eager:context", ], @@ -3224,6 +3227,18 @@ py_test( ) py_test( + name = "util_serialization_test", + size = "small", + srcs = ["util/serialization_test.py"], + main = "util/serialization_test.py", + srcs_version = "PY2AND3", + deps = [ + ":client_testlib", + ":util", + ], +) + +py_test( name = "future_api_test", size = "small", srcs = ["util/future_api_test.py"], |