Searched refs:ret_input_node (Results 1 – 1 of 1) sorted by relevance
52 Node* ret_input_node; in ComputeShortCircuitIndices() local53 TF_RETURN_IF_ERROR(ret_node->input_node(0, &ret_input_node)); in ComputeShortCircuitIndices()55 while (ret_input_node->def().op() == "Identity") { in ComputeShortCircuitIndices()56 TF_RETURN_IF_ERROR(ret_input_node->input_node(0, &ret_input_node)); in ComputeShortCircuitIndices()59 if (ret_input_node->def().op() == FunctionLibraryDefinition::kArgOp) { in ComputeShortCircuitIndices()61 GetNodeAttr(ret_input_node->def(), "index", &((*indices)[i]))); in ComputeShortCircuitIndices()