Home
last modified time | relevance | path

Searched refs:element_prefix (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc294 std::string element_prefix = in SaveInternal() local
297 WriteStatusLocked(writer, element_prefix, result.status)); in SaveInternal()
298 TF_RETURN_IF_ERROR(writer->WriteScalar(element_prefix, kSize, in SaveInternal()
302 element_prefix, absl::StrCat(kComponent, "[", j, "]"), in SaveInternal()
307 writer->WriteScalar(element_prefix, kEndOfInput, "")); in SaveInternal()
325 std::string element_prefix = in RestoreInternal() local
328 ReadStatusLocked(reader, element_prefix, &result.status)); in RestoreInternal()
332 TF_RETURN_IF_ERROR(reader->ReadScalar(element_prefix, kSize, &size)); in RestoreInternal()
336 element_prefix, ",", kSize, ": ", size, in RestoreInternal()
344 element_prefix, absl::StrCat(kComponent, "[", j, "]"), in RestoreInternal()
[all …]
Ddataset_utils.cc63 std::string element_prefix = absl::StrCat(key_prefix, "::", i); in WriteElementsToCheckpoint() local
65 writer->WriteScalar(element_prefix, kNumComponents, element.size())); in WriteElementsToCheckpoint()
68 element_prefix, absl::StrCat(kComponent, "[", j, "]"), element[j])); in WriteElementsToCheckpoint()
82 std::string element_prefix = absl::StrCat(key_prefix, "::", i); in ReadElementsFromCheckpoint() local
85 reader->ReadScalar(element_prefix, kNumComponents, &num_components)); in ReadElementsFromCheckpoint()
92 element_prefix, absl::StrCat(kComponent, "[", j, "]"), in ReadElementsFromCheckpoint()