Searched refs:HostGraphControlRetMapping (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass.cc | 46 std::optional<string> HostGraphControlRetMapping(const Node* n) { in HostGraphControlRetMapping() function
655 HostGraphControlRetMapping, &rewritten_body_function_def)); in PostprocessLiftedArgsForWhile()
681 HostGraphControlRetMapping, &rewritten_cond_function_def)); in PostprocessLiftedArgsForWhile()
810 HostGraphControlRetMapping, &rewritten_then_branch_function_def)); in PostprocessLiftedArgsForIf()
820 HostGraphControlRetMapping, &rewritten_else_branch_function_def)); in PostprocessLiftedArgsForIf()
884 HostGraphControlRetMapping, in PostprocessLiftedArgsForCall()
1399 *g, func_name, HostGraphControlRetMapping, &replace_fdef)); in ReplaceKeyPlaceholderWithArgNode()
1626 HostGraphControlRetMapping, in RewriteHostWhileLoopCond()
1684 HostGraphControlRetMapping, in RewriteHostWhileLoopBody()
1796 HostGraphControlRetMapping, in BuildHostGraphForFuncCallNode()
[all …]
|