[tf.data] Fix a possible collision of performance modeling identifiers.

PiperOrigin-RevId: 245790338
2019-04-29 11:42:05 -07:00 · 2019-04-29 11:42:05 -07:00 · d8c82221f0
commit d8c82221f0
parent eddf4efbc5
1 changed files with 3 additions and 2 deletions
--- a/tensorflow/core/kernels/data/experimental/group_by_window_dataset_op.cc
+++ b/tensorflow/core/kernels/data/experimental/group_by_window_dataset_op.cc
@ -471,8 +471,9 @@ class GroupByWindowDatasetOp : public UnaryDatasetOpKernel {
            GetDatasetFromVariantTensor(return_values[0], &returned_dataset));

        // Create an iterator for the dataset that was returned by `f`.
-        return returned_dataset->MakeIterator(ctx, prefix(),
-                                              &current_group_iterator_);
+        return returned_dataset->MakeIterator(
+            ctx, strings::StrCat(prefix(), "::Reduce"),
+            &current_group_iterator_);
      }

      mutex mu_;