Searched refs:function_fbody (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 2403 std::unique_ptr<FunctionBody> function_fbody; in LiftOutsideCompilationOnlyArgs() local
2404 TF_ASSIGN_OR_RETURN(function_fbody, in LiftOutsideCompilationOnlyArgs()
2408 fld->UniqueFunctionName(function_fbody->fdef.signature().name()); in LiftOutsideCompilationOnlyArgs()
2410 *function_fbody, flr, fld, lifted_arg_count, new_func_name, in LiftOutsideCompilationOnlyArgs()
2423 std::unique_ptr<FunctionBody> function_fbody; in LiftOutsideCompilationOnlyArgs() local
2427 &function_fbody)); in LiftOutsideCompilationOnlyArgs()
2430 fld->UniqueFunctionName(function_fbody->fdef.signature().name()); in LiftOutsideCompilationOnlyArgs()
2432 *function_fbody, flr, fld, lifted_arg_count, new_func_name, in LiftOutsideCompilationOnlyArgs()
2455 const FunctionBody* function_fbody = flr->GetFunctionBody(handle); in LiftOutsideCompilationOnlyArgs() local
2457 *function_fbody, flr, fld, lifted_arg_count, new_func_name, in LiftOutsideCompilationOnlyArgs()
|