From 8f54070dd1abe9453f9384cdb37fad89da16210a Mon Sep 17 00:00:00 2001 From: Jiri Simsa Date: Fri, 28 Aug 2020 16:50:45 -0700 Subject: [PATCH] [tf.data] fixing disable intra op parallelism PiperOrigin-RevId: 329033197 Change-Id: I2985864483fcdc4249672df7fd19d8182509e14a --- .../optimizers/data/disable_intra_op_parallelism.cc | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/tensorflow/core/grappler/optimizers/data/disable_intra_op_parallelism.cc b/tensorflow/core/grappler/optimizers/data/disable_intra_op_parallelism.cc index 251a66fc6f3..52f1ba59b32 100644 --- a/tensorflow/core/grappler/optimizers/data/disable_intra_op_parallelism.cc +++ b/tensorflow/core/grappler/optimizers/data/disable_intra_op_parallelism.cc @@ -96,12 +96,7 @@ Status DisableIntraOpParallelism::OptimizeAndCollectStats( if (last_node->attr().find(attr) != last_node->attr().end()) { graph_utils::CopyAttribute(attr, *last_node, &insert_node); } else { - auto inferred_attr = strings::StrCat("T", attr); - if (last_node->attr().find(inferred_attr) != last_node->attr().end()) { - graph_utils::CopyAttribute(inferred_attr, *last_node, &insert_node); - } else { - return Status::OK(); - } + return Status::OK(); } }