Searched refs:cond_fbody (Results 1 – 2 of 2) sorted by relevance
1783 std::unique_ptr<FunctionBody> cond_fbody; in FindArgsToLiftForWhileNode() local1785 *cond_fdef, AttrSlice(&cond_func.attr()), fld, &cond_fbody)); in FindArgsToLiftForWhileNode()1786 for (int i = 0; i < cond_fbody->arg_nodes.size(); i++) { in FindArgsToLiftForWhileNode()1787 const Node* arg_node = cond_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode()2151 TF_ASSIGN_OR_RETURN(std::unique_ptr<FunctionBody> cond_fbody, in LiftOutsideCompilationOnlyArgsFromWhileNode()2155 cond_fbody.get())); in LiftOutsideCompilationOnlyArgsFromWhileNode()2158 TF_RETURN_IF_ERROR(GraphToFunctionDef(*(cond_fbody->graph), in LiftOutsideCompilationOnlyArgsFromWhileNode()2159 cond_fbody->fdef.signature().name(), in LiftOutsideCompilationOnlyArgsFromWhileNode()2161 TF_RETURN_IF_ERROR(fld->ReplaceFunction(cond_fbody->fdef.signature().name(), in LiftOutsideCompilationOnlyArgsFromWhileNode()
1564 std::unique_ptr<FunctionBody> cond_fbody; in RewriteHostWhileLoopCond() local1567 fld, &cond_fbody)); in RewriteHostWhileLoopCond()1568 Graph* cond_graph = cond_fbody->graph; in RewriteHostWhileLoopCond()