Commit Graph

11 Commits

Author SHA1 Message Date
Gunhan Gulsoy
cabb1ed2cf Reduce dependence from tensorflow/c to tensorflow/core/lib.
Use the migrated headers under tensorflow/core/platform.

PiperOrigin-RevId: 304194185
Change-Id: I476004f49e554b07f5670b7f5b0c3c61a5ddc839
2020-04-01 08:56:02 -07:00
Gaurav Jain
3afb32ce1f Use TF_Status* SWIG typemap
This avoids the need to use raise_exception_on_not_ok_status
which has been removed from the v2 API.

PiperOrigin-RevId: 244049607
2019-04-17 13:06:01 -07:00
A. Unique TensorFlower
eece29de7b Removed redundant std::string -> string conversions.
PiperOrigin-RevId: 210596417
2018-08-28 13:41:11 -07:00
Peter Hawkins
5e9e6967b4 Replaced calls to tensorflow::StringPiece::ToString with std::string conversions.
That is, instances of sp.ToString() are replaced with std::string(sp).

This will allow tensorflow::StringPiece::ToString to be removed, which is necessary before it can be replaced with absl::string_view.

PiperOrigin-RevId: 195162126
2018-05-02 16:22:41 -07:00
A. Unique TensorFlower
a8c5d5fe01 Expose data type information in checkpoint reader.
PiperOrigin-RevId: 171147196
2017-10-05 07:01:19 -07:00
A. Unique TensorFlower
220515bffd Replace owning raw pointers with unique pointers
PiperOrigin-RevId: 171132628
2017-10-05 03:49:51 -07:00
A. Unique TensorFlower
0656952250 Fix memory leakage of CheckpointReader
Change: 146955950
2017-02-08 14:30:40 -08:00
Zongheng Yang
1e324a0f2a checkpoint_reader: fix VarToShapeMap V2 impl.
This change makes it adhere to the original semantics: all slices of a
partitioned tensor are grouped under one entry.
Change: 136229541
2016-10-14 22:18:59 -07:00
Zongheng Yang
cae3713cd4 tensor_bundle: tighten Lookup() semantics, add LookupDtypeShape().
This fixes a subtle allocation issue in CheckpointReader.
Change: 135503615
2016-10-07 12:33:03 -07:00
Zongheng Yang
ecdf0b202a 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
2016-09-28 09:35:36 -07:00
A. Unique TensorFlower
788f359b72 Move C API files (and related files used by SWIG wrappers)
to new tensorflow/c directory.
Change: 128855990
2016-07-29 16:03:48 -07:00