Searched refs:body_function_body (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 600 std::unique_ptr<FunctionBody> body_function_body; in PostprocessLiftedArgsForWhile() local
603 &body_function_body)); in PostprocessLiftedArgsForWhile()
605 int original_arg_count = body_function_body->arg_nodes.size(); in PostprocessLiftedArgsForWhile()
610 *body_function_body, outside_compilation_attr_to_node)); in PostprocessLiftedArgsForWhile()
641 *body_function_body, i, data_types[i])); in PostprocessLiftedArgsForWhile()
644 AddMatchingRetvalNode(*body_function_body, i, data_types[i], arg_node)); in PostprocessLiftedArgsForWhile()
647 *body_function_body, original_arg_count, i, lifted_arg_nodes, arg_node); in PostprocessLiftedArgsForWhile()
654 *body_function_body->graph, new_body_function_name, in PostprocessLiftedArgsForWhile()
|