Searched refs:body_fbody (Results 1 – 2 of 2) sorted by relevance
1803 std::unique_ptr<FunctionBody> body_fbody; in FindArgsToLiftForWhileNode() local1805 *body_fdef, AttrSlice(&body_func.attr()), fld, &body_fbody)); in FindArgsToLiftForWhileNode()1806 for (int i = 0; i < body_fbody->ret_nodes.size(); i++) { in FindArgsToLiftForWhileNode()1807 const Node* node = body_fbody->ret_nodes[i]; in FindArgsToLiftForWhileNode()1811 if (node != body_fbody->arg_nodes[i]) { in FindArgsToLiftForWhileNode()1818 for (int i = 0; i < body_fbody->arg_nodes.size(); i++) { in FindArgsToLiftForWhileNode()1819 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode()1829 for (int i = 0; i < body_fbody->arg_nodes.size(); i++) { in FindArgsToLiftForWhileNode()1830 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode()2166 TF_ASSIGN_OR_RETURN(std::unique_ptr<FunctionBody> body_fbody, in LiftOutsideCompilationOnlyArgsFromWhileNode()[all …]
1640 std::unique_ptr<FunctionBody> body_fbody; in RewriteHostWhileLoopBody() local1644 fld, &body_fbody)); in RewriteHostWhileLoopBody()1645 Graph* body_graph = body_fbody->graph; in RewriteHostWhileLoopBody()