Searched refs:ReadElementsFromCheckpoint (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | serialization_utils.h | 28 Status ReadElementsFromCheckpoint(IteratorContext* ctx,
|
| H A D | serialization_utils_test.cc | 113 TF_ASSERT_OK(ReadElementsFromCheckpoint(ctx->iter_ctx(), &reader, test_prefix, in TEST()
|
| H A D | serialization_utils.cc | 105 Status ReadElementsFromCheckpoint(IteratorContext* ctx, in ReadElementsFromCheckpoint() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | cache_dataset_ops.cc | 840 ReadElementsFromCheckpoint(ctx, reader, prefix(), &temp_cache)); in RestoreInternal()
911 ReadElementsFromCheckpoint(ctx, reader, prefix(), &temp_cache_)); in RestoreInternal()
|
| H A D | shuffle_dataset_op.cc | 335 ReadElementsFromCheckpoint(ctx, reader, prefix(), buffer_.get())); in RestoreInternal()
|