diff options
| author |  Shivani Agrawal <shivaniagrawal@google.com> | 2018-08-01 14:52:59 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-01 15:01:24 -0700 |
| commit | 52dc7286bda07a53b4bc6e5ca17ff22fc5d72af5 (patch) | |
| tree | e4d230a155a10ef237b8fc05e531f98f39349d56 /tensorflow/python/BUILD | |
| parent | a28ad4b26dbb8cb1e9cf2135f72f3f55ffabf037 (diff) | |
[Checkpointable] Make Iterator checkpointable.
Use object-based save/restore to make dataset/iterator checkpointable in both graph as well as eager mode.
PiperOrigin-RevId: 206998349
Diffstat (limited to 'tensorflow/python/BUILD')
| -rw-r--r-- | tensorflow/python/BUILD | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index d35731d3cd..2ccaae4dcf 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3216,6 +3216,7 @@ py_library( # The following targets have their own build rules (same name as the # file): "training/saveable_object.py", + "training/saver.py", "training/training_util.py", ], ), @@ -3247,6 +3248,7 @@ py_library( ":random_ops", ":resource_variable_ops", ":resources", + "saver", ":saveable_object", ":sdca_ops", ":sparse_ops", @@ -3278,6 +3280,40 @@ py_library( ) py_library( + name = "saver", + srcs = ["training/saver.py"], + srcs_version = "PY2AND3", + deps = [ + ":array_ops", + ":constant_op", + ":control_flow_ops", + ":device", + ":errors", + ":framework", + ":framework_ops", + ":io_ops", + ":io_ops_gen", + ":lib", + ":platform", + ":protos_all_py", + ":pywrap_tensorflow", + ":resource_variable_ops", + ":saveable_object", + ":session", + ":state_ops", + ":string_ops", + ":training_util", + ":util", + ":variables", + "//tensorflow/core:protos_all_py", + "//tensorflow/python/eager:context", + "//tensorflow/python/training/checkpointable:base", + "//third_party/py/numpy", + "@six_archive//:six", + ], +) + +py_library( name = "device_util", srcs = ["training/device_util.py"], srcs_version = "PY2AND3", |