Searched refs:CodeKey (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | window_dataset_op.cc | 338 CodeKey(index), static_cast<int64_t>(status.code()))); in WriteStatusLocked()
349 TF_RETURN_IF_ERROR(reader->ReadScalar(CodeKey(index), &code_int)); in ReadStatusLocked()
363 string CodeKey(size_t index) { in CodeKey() function in tensorflow::data::WindowDatasetOp::Dataset::Iterator
|
| H A D | prefetch_dataset_op.cc | 526 writer->WriteScalar(absl::StrCat(prefix(), "::", index), CodeKey(), in WriteStatus()
540 CodeKey(), &code_int)); in ReadStatus()
555 string CodeKey() { return absl::StrCat(kStatus, kCodeSuffix); } in CodeKey() function in tensorflow::data::PrefetchDatasetOp::Dataset::Iterator
|
| H A D | parallel_interleave_dataset_op.cc | 1239 iterator_name, CodeKey(idx), static_cast<int64_t>(status.code()))); in WriteStatusLocked()
1252 reader->ReadScalar(iterator_name, CodeKey(idx), &code_int)); in ReadStatusLocked()
1266 string CodeKey(size_t idx) { in CodeKey() function in tensorflow::data::ParallelInterleaveDatasetOp::Dataset::ParallelInterleaveIterator
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | parse_example_dataset_op.cc | 856 CodeKey(index), static_cast<int64_t>(status.code()))); in WriteStatusLocked()
868 TF_RETURN_IF_ERROR(reader->ReadScalar(CodeKey(index), &code_int)); in ReadStatusLocked()
882 string CodeKey(size_t index) { in CodeKey() function in tensorflow::data::experimental::__anon4014e9290111::ParseExampleDatasetOp::Dataset::Iterator
|
| H A D | snapshot_dataset_op.cc | 1591 CodeKey(index), static_cast<int64_t>(status.code()))); in WriteStatus()
1602 TF_RETURN_IF_ERROR(reader->ReadScalar(CodeKey(index), &code_int)); in ReadStatus()
1616 string CodeKey(size_t index) { in CodeKey() function in tensorflow::data::experimental::__anon34efa0f30311::SnapshotDatasetOp::Dataset::Iterator::SnapshotReaderIterator
|