Searched refs:element_prefix (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | parallel_map_dataset_op.cc | 294 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 …]
|
D | dataset_utils.cc | 63 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()
|