Searched refs:retval_node (Results 1 – 2 of 2) sorted by relevance
130 Node* retval_node = nullptr; in AddRetvalNodes() local138 .Finalize(graph, &retval_node)); in AddRetvalNodes()139 retval_nodes->insert(retval_node); in AddRetvalNodes()
180 TF_ASSIGN_OR_RETURN(Node * retval_node, BuildRetvalNode(output, dtype, i)); in BuildLoopBody()184 output->AddEdge(arg_node, 0, retval_node, 0); in BuildLoopBody()200 node_map[arg.next_iteration->id()] = retval_node; in BuildLoopBody()