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

Made sure that both queue shapes and types are updated in one single pass instead of two.

Browse Source 
Change: 154960772
This commit is contained in:
Benoit Steiner 2017-05-03 07:52:23 -08:00 committed by TensorFlower Gardener
parent 562136cf7f
commit 27aaf4a653
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/grappler/costs/graph_properties.cc
View File

@ -84,7 +84,7 @@ Status GraphProperties::InferStatically() {
}
}
}
if (qctx->set_output_handle_dtype(0, queue_type) ||
if (qctx->set_output_handle_dtype(0, queue_type) |
qctx->MergeOutputHandleShape(0, queue_shp)) {
new_shapes.push(qnode);
}