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);