Home
last modified time | relevance | path

Searched refs:WriteElementsToCheckpoint (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/data/
Dserialization_utils.h37 Status WriteElementsToCheckpoint(
Dserialization_utils_test.cc104 TF_ASSERT_OK(WriteElementsToCheckpoint(&writer, test_prefix, elements)); in TEST()
Dserialization_utils.cc132 Status WriteElementsToCheckpoint( in WriteElementsToCheckpoint() function
/external/tensorflow/tensorflow/core/kernels/data/
Dcache_dataset_ops.cc827 WriteElementsToCheckpoint(writer, prefix(), cache_->data())); in SaveInternal()
901 WriteElementsToCheckpoint(writer, prefix(), temp_cache_)); in SaveInternal()
Dshuffle_dataset_op.cc278 TF_RETURN_IF_ERROR(WriteElementsToCheckpoint(writer, prefix(), *buffer_)); in SaveInternal()