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

More informative node names in IF lowering.

Browse Source 
(Looks like the original was just a typo.)

PiperOrigin-RevId: 241945725
This commit is contained in:
Alexey Radul 2019-04-04 09:36:28 -07:00 committed by TensorFlower Gardener
parent fa5ad62d32
commit 30bcf6ae7c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/common_runtime/lower_if_op.cc
View File

@ -206,7 +206,7 @@ Status CondBuilder::AddOutputs() {
outputs_.resize(merges.size()); outputs_.resize(merges.size());
for (int i = 0; i < then_call_node_->num_outputs(); ++i) { for (int i = 0; i < then_call_node_->num_outputs(); ++i) {
TF_RETURN_IF_ERROR( TF_RETURN_IF_ERROR(
NodeBuilder(graph_->NewName("output"), "Merge", graph_->op_registry(), NodeBuilder(NewName("output"), "Merge", graph_->op_registry(),
&debug_info_) &debug_info_)
.Input({NodeOut(then_call_node_, i), NodeOut(else_call_node_, i)}) .Input({NodeOut(then_call_node_, i), NodeOut(else_call_node_, i)})
.Device(if_op_->requested_device()) .Device(if_op_->requested_device())
@ -224,7 +224,7 @@ Status CondBuilder::AddOutputs() {
// //
// We will use this node to rewrite outgoing control edges from lowered 'If' // We will use this node to rewrite outgoing control edges from lowered 'If'
// node. All data edges will read tensors directly from Merge nodes. // node. All data edges will read tensors directly from Merge nodes.
TF_RETURN_IF_ERROR(NodeBuilder(graph_->NewName("branch_executed"), "Merge", TF_RETURN_IF_ERROR(NodeBuilder(NewName("branch_executed"), "Merge",
graph_->op_registry(), &debug_info_) graph_->op_registry(), &debug_info_)
.Input({pivot_t_, pivot_f_}) .Input({pivot_t_, pivot_f_})
.ControlInputs({then_call_node_, else_call_node_}) .ControlInputs({then_call_node_, else_call_node_})