Searched refs:pred_edge (Results 1 – 2 of 2) sorted by relevance
121 const Edge* pred_edge; in GetSwitchPredicate() local122 TF_RETURN_IF_ERROR(switch_node.input_edge(1, &pred_edge)); in GetSwitchPredicate()125 while (pred_edge->src()->IsIdentity()) { in GetSwitchPredicate()126 TF_RETURN_IF_ERROR(pred_edge->src()->input_edge(0, &pred_edge)); in GetSwitchPredicate()128 *pred = OutputTensor(pred_edge->src(), pred_edge->src_output()); in GetSwitchPredicate()
791 const Edge* pred_edge; in HandleSwitch() local792 TF_RETURN_IF_ERROR(n->input_edge(1, &pred_edge)); in HandleSwitch()796 pred_edge->src(), pred_edge->src_output(), in HandleSwitch()