Home
last modified time | relevance | path

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

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