diff options
author | Zongheng Yang <zongheng@google.com> | 2016-09-28 08:26:07 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-28 09:35:36 -0700 |
commit | ecdf0b202a2bfcff7985e62da727397bd8c67a91 (patch) | |
tree | b81d13a2f910cb0f34794b2e46ba08f6415124e1 /tensorflow/c/BUILD | |
parent | 63a7a30e6bd091f87be1de2305c6d882d68ba6a8 (diff) |
TF Checkpoint V2: make CheckpointReader work with the V2 format.
If the same checkpoint prefix identifies both a V1 checkpoint and a V2
checkpoint on disk, the V2 version takes priority -- which matches the same
behavior as the RestoreV2 op.
Typical usage:
$ bazel run tensorflow/python/tools:inspect_checkpoint -- --file_name=<V2 ckpt prefix>
Other changes: add DebugString() and Contains() to BundleReader.
Change: 134543092
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 7f86d345d7..7afac44452 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -51,6 +51,7 @@ tf_cuda_library( ":tf_status_helper", "//tensorflow/core:framework", "//tensorflow/core:lib", + "//tensorflow/core/util/tensor_bundle", ], ) |