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

[tf.data] fixing disable intra op parallelism

Browse Source 
PiperOrigin-RevId: 329033197
Change-Id: I2985864483fcdc4249672df7fd19d8182509e14a
This commit is contained in:
Jiri Simsa 2020-08-28 16:50:45 -07:00 committed by TensorFlower Gardener
parent 93d68885c4
commit 8f54070dd1
1 changed files with 1 additions and 6 deletions

7
tensorflow/core/grappler/optimizers/data/disable_intra_op_parallelism.cc
View File

@ -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();
}
}