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