Searched refs:if_node_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 371 Node* if_node_ = nullptr; member in tensorflow::functionalize_cond::Conditional
798 TF_ASSIGN_OR_RETURN(if_node_, in BuildIfNode()
807 VLOG(2) << "AddInputEdges for " << if_node_->name(); in AddInputEdges()
822 graph->AddEdge(iter->second.node, iter->second.index, if_node_, index++); in AddInputEdges()
825 if_node_, index++); in AddInputEdges()
830 graph->AddControlEdge(arg.src, if_node_); in AddInputEdges()
832 graph->AddEdge(arg.src, arg.src_output, if_node_, index++); in AddInputEdges()
836 graph->AddControlEdge(n, if_node_); in AddInputEdges()
844 VLOG(2) << "AddOutputEdges for " << if_node_->name(); in AddOutputEdges()
847 TF_RETURN_IF_ERROR(parent_->AddIdentityNode(node, if_node_, i)); in AddOutputEdges()
[all …]
|