From 7a33771b76da4e19640ae149e283fdb43102983c Mon Sep 17 00:00:00 2001 From: Jiri Simsa Date: Fri, 12 Jun 2020 14:07:25 -0700 Subject: [PATCH] [tf.data] Remove extra "tf_" prefix from thread names of parallel interleave threads. PiperOrigin-RevId: 316173922 Change-Id: I992af699b647347797d010ec1ce5161f81fd5960 --- .../data/experimental/parallel_interleave_dataset_op.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/core/kernels/data/experimental/parallel_interleave_dataset_op.cc b/tensorflow/core/kernels/data/experimental/parallel_interleave_dataset_op.cc index c4b34b5163c..2167c5d9b98 100644 --- a/tensorflow/core/kernels/data/experimental/parallel_interleave_dataset_op.cc +++ b/tensorflow/core/kernels/data/experimental/parallel_interleave_dataset_op.cc @@ -70,8 +70,8 @@ constexpr char kInterleaveIndices[] = "interleave_indices"; constexpr char kStagingSize[] = "staging_size"; constexpr char kStagingIndices[] = "staging_indices"; constexpr char kWorkerThreadsRunning[] = "worker_threads_running"; -constexpr char kTFDataParallelInterleaveWorker[] = - "tf_data_parallel_interleave_worker"; +constexpr char kDataParallelInterleaveWorker[] = + "data_parallel_interleave_worker"; constexpr char kWorker[] = "worker"; constexpr char kInputSize[] = "input_size"; constexpr char kInput[] = "input"; @@ -544,7 +544,7 @@ class ParallelInterleaveDatasetOp::Dataset : public DatasetBase { for (size_t i = 0; i < dataset()->num_threads(); ++i) { std::shared_ptr new_ctx(new IteratorContext(*ctx)); worker_threads_.emplace_back(ctx->StartThread( - strings::StrCat(kTFDataParallelInterleaveWorker, "_", i), + strings::StrCat(kDataParallelInterleaveWorker, "_", 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)); std::shared_ptr new_ctx(new IteratorContext(*ctx)); worker_threads_.push_back(ctx->StartThread( - strings::StrCat(kTFDataParallelInterleaveWorker, "_", i), + strings::StrCat(kDataParallelInterleaveWorker, "_", i), [this, new_ctx, i]() { WorkerThread(new_ctx, i); })); if (i < dataset()->cycle_length_) { interleave_indices_.push_back(i);