Searched refs:rewritten_fdef (Results 1 – 2 of 2) sorted by relevance
1889 FunctionDef rewritten_fdef; in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef() local1891 *(fbody.graph), fbody.fdef.signature().name(), &rewritten_fdef)); in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()1893 rewritten_fdef.mutable_signature()->set_name(*new_func_name); in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()1894 TF_RETURN_IF_ERROR(fld->AddFunctionDef(rewritten_fdef)); in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()1897 fld->ReplaceFunction(fbody.fdef.signature().name(), rewritten_fdef)); in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()2316 FunctionDef rewritten_fdef; in LiftOutsideCompilationOnlyArgsFromCallNode() local2318 *(fbody->graph), fbody->fdef.signature().name(), &rewritten_fdef)); in LiftOutsideCompilationOnlyArgsFromCallNode()2321 rewritten_fdef.mutable_signature()->set_name(new_func_name); in LiftOutsideCompilationOnlyArgsFromCallNode()2322 TF_RETURN_IF_ERROR(fld->AddFunctionDef(rewritten_fdef)); in LiftOutsideCompilationOnlyArgsFromCallNode()
876 FunctionDef rewritten_fdef; in PostprocessLiftedArgsForCall() local879 &rewritten_fdef)); in PostprocessLiftedArgsForCall()882 rewritten_fdef.mutable_signature()->set_name(new_function_name); in PostprocessLiftedArgsForCall()883 TF_RETURN_IF_ERROR(fld->AddFunctionDef(rewritten_fdef)); in PostprocessLiftedArgsForCall()