Searched refs:ret_input_node (Results 1 – 1 of 1) sorted by relevance
195 Node* ret_input_node; in CreateShortCircuitInfo() local196 TF_RETURN_IF_ERROR(ret_node->input_node(0, &ret_input_node)); in CreateShortCircuitInfo()198 while (ret_input_node->def().op() == "Identity") { in CreateShortCircuitInfo()199 TF_RETURN_IF_ERROR(ret_input_node->input_node(0, &ret_input_node)); in CreateShortCircuitInfo()202 if (ret_input_node->def().op() == FunctionLibraryDefinition::kArgOp) { in CreateShortCircuitInfo()204 GetNodeAttr(ret_input_node->def(), "index", &(indices[i]))); in CreateShortCircuitInfo()