experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[tf.data] Remove extra "tf_" prefix from thread names of parallel interleave threads. 

PiperOrigin-RevId: 316173922
Change-Id: I992af699b647347797d010ec1ce5161f81fd5960
This commit is contained in:
Jiri Simsa 2020-06-12 14:07:25 -07:00 committed by TensorFlower Gardener
parent 3a2d113417
commit 7a33771b76
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/kernels/data/experimental/parallel_interleave_dataset_op.cc
View File

@ -70,8 +70,8 @@ constexpr char kInterleaveIndices[] = "interleave_indices";
constexpr char kStagingSize[] = "staging_size"; constexpr char kStagingSize[] = "staging_size";
constexpr char kStagingIndices[] = "staging_indices"; constexpr char kStagingIndices[] = "staging_indices";
constexpr char kWorkerThreadsRunning[] = "worker_threads_running"; constexpr char kWorkerThreadsRunning[] = "worker_threads_running";
constexpr char kTFDataParallelInterleaveWorker[] = constexpr char kDataParallelInterleaveWorker[] =
"tf_data_parallel_interleave_worker"; "data_parallel_interleave_worker";
constexpr char kWorker[] = "worker"; constexpr char kWorker[] = "worker";
constexpr char kInputSize[] = "input_size"; constexpr char kInputSize[] = "input_size";
constexpr char kInput[] = "input"; constexpr char kInput[] = "input";
@ -544,7 +544,7 @@ class ParallelInterleaveDatasetOp::Dataset : public DatasetBase {
for (size_t i = 0; i < dataset()->num_threads(); ++i) { for (size_t i = 0; i < dataset()->num_threads(); ++i) {
std::shared_ptr<IteratorContext> new_ctx(new IteratorContext(*ctx)); std::shared_ptr<IteratorContext> new_ctx(new IteratorContext(*ctx));
worker_threads_.emplace_back(ctx->StartThread( worker_threads_.emplace_back(ctx->StartThread(
strings::StrCat(kTFDataParallelInterleaveWorker, "_", i), strings::StrCat(kDataParallelInterleaveWorker, "_", i),
[this, new_ctx, i]() { WorkerThread(new_ctx, i); })); [this, new_ctx, i]() { WorkerThread(new_ctx, i); }));
} }
} }
@ -655,7 +655,7 @@ class ParallelInterleaveDatasetOp::Dataset : public DatasetBase {
workers_[i].SetInputs(s, std::move(args)); workers_[i].SetInputs(s, std::move(args));
std::shared_ptr<IteratorContext> new_ctx(new IteratorContext(*ctx)); std::shared_ptr<IteratorContext> new_ctx(new IteratorContext(*ctx));
worker_threads_.push_back(ctx->StartThread( worker_threads_.push_back(ctx->StartThread(
strings::StrCat(kTFDataParallelInterleaveWorker, "_", i), strings::StrCat(kDataParallelInterleaveWorker, "_", i),
[this, new_ctx, i]() { WorkerThread(new_ctx, i); })); [this, new_ctx, i]() { WorkerThread(new_ctx, i); }));
if (i < dataset()->cycle_length_) { if (i < dataset()->cycle_length_) {
interleave_indices_.push_back(i); interleave_indices_.push_back(i);