Searched refs:pred_edge (Results 1 – 2 of 2) sorted by relevance
935 const Edge* pred_edge; in HandleSwitch() local936 TF_RETURN_IF_ERROR(n->input_edge(1, &pred_edge)); in HandleSwitch()941 pred_edge->src(), pred_edge->src_output(), in HandleSwitch()961 pred_edge->src(), pred_edge->src_output(), in HandleSwitch()
125 const Edge* pred_edge; in GetSwitchPredicate() local126 TF_RETURN_IF_ERROR(switch_node.input_edge(1, &pred_edge)); in GetSwitchPredicate()129 while (pred_edge->src()->IsIdentity()) { in GetSwitchPredicate()130 TF_RETURN_IF_ERROR(pred_edge->src()->input_edge(0, &pred_edge)); in GetSwitchPredicate()132 *pred = OutputTensor(pred_edge->src(), pred_edge->src_output()); in GetSwitchPredicate()