Searched refs:if_node_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 384 Node* if_node_ = nullptr; member in tensorflow::functionalize_cond::Conditional
839 TF_ASSIGN_OR_RETURN(if_node_, in BuildIfNode()
848 VLOG(2) << "AddInputEdges for " << if_node_->name(); in AddInputEdges()
863 graph->AddEdge(iter->second.node, iter->second.index, if_node_, index++); in AddInputEdges()
866 if_node_, index++); in AddInputEdges()
871 graph->AddControlEdge(arg.src, if_node_); in AddInputEdges()
873 graph->AddEdge(arg.src, arg.src_output, if_node_, index++); in AddInputEdges()
877 graph->AddControlEdge(n, if_node_); in AddInputEdges()
885 VLOG(2) << "AddOutputEdges for " << if_node_->name(); in AddOutputEdges()
888 TF_RETURN_IF_ERROR(parent_->AddIdentityNode(node, if_node_, i)); in AddOutputEdges()
[all …]
|