Searched refs:CodeKey (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | prefetch_dataset_op.cc | 330 CodeKey(index), static_cast<int64>(status.code()))); in WriteStatus()
341 TF_RETURN_IF_ERROR(reader->ReadScalar(CodeKey(index), &code_int)); in ReadStatus()
355 string CodeKey(size_t index) { in CodeKey() function in tensorflow::data::PrefetchDatasetOp::Dataset::Iterator
|
| D | parallel_map_iterator.cc | 344 writer->WriteScalar(CodeKey(index), static_cast<int64>(status.code()))); in WriteStatusLocked()
355 TF_RETURN_IF_ERROR(reader->ReadScalar(CodeKey(index), &code_int)); in ReadStatusLocked()
369 string CodeKey(size_t index) { in CodeKey() function in tensorflow::data::__anond6ede6490111::ParallelMapIterator
|
| D | window_dataset_op.cc | 324 CodeKey(index), static_cast<int64>(status.code()))); in WriteStatusLocked()
335 TF_RETURN_IF_ERROR(reader->ReadScalar(CodeKey(index), &code_int)); in ReadStatusLocked()
349 string CodeKey(size_t index) { in CodeKey() function in tensorflow::data::__anon256b48110111::WindowDatasetOp::Dataset::Iterator
|
| D | parallel_interleave_dataset_op.cc | 669 CodeKey(key_prefix, idx), static_cast<int64>(status.code()))); in WriteStatusLocked()
682 reader->ReadScalar(CodeKey(key_prefix, idx), &code_int)); in ReadStatusLocked()
696 string CodeKey(const string& key_prefix, size_t idx) { in CodeKey() function in tensorflow::data::__anon6594f9940111::ParallelInterleaveDatasetOp::Dataset::ParallelInterleaveIterator
|