Searched refs:original_arg_count (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 501 const int original_arg_count, const int arg_to_input_edge_offset, in AddEdgesFromOutsideCompilationNodes() argument
505 for (int i = original_arg_count, end = data_types.size(); i < end; i++) { in AddEdgesFromOutsideCompilationNodes()
507 outside_compilation_nodes[i - original_arg_count]; in AddEdgesFromOutsideCompilationNodes()
548 const FunctionBody& function_body, const int original_arg_count, in ReplaceLiftedArgNodePlaceholderWithArg() argument
551 Node* lifted_arg_node = lifted_arg_nodes[arg_idx - original_arg_count]; in ReplaceLiftedArgNodePlaceholderWithArg()
605 int original_arg_count = body_function_body->arg_nodes.size(); in PostprocessLiftedArgsForWhile() local
625 AddEdgesFromOutsideCompilationNodes(original_arg_count, in PostprocessLiftedArgsForWhile()
638 for (int i = original_arg_count, end = data_types.size(); i < end; i++) { in PostprocessLiftedArgsForWhile()
647 *body_function_body, original_arg_count, i, lifted_arg_nodes, arg_node); in PostprocessLiftedArgsForWhile()
670 for (int i = original_arg_count, end = data_types.size(); i < end; i++) { in PostprocessLiftedArgsForWhile()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | stack_ops_decomposition.cc | 125 int64_t original_arg_count = new_input_types.size(); in ModifyFunctionSignature() local
126 for (int64_t i = 0; i < original_arg_count; ++i) { in ModifyFunctionSignature()
138 handle_new_size_vars(func.getArguments().drop_front(original_arg_count)); in ModifyFunctionSignature()
|
| D | tensor_list_ops_decomposition.cc | 94 int64_t original_arg_count = new_input_types.size(); in ModifyFunctionSignature() local
95 for (int64_t i = 0; i < original_arg_count; ++i) { in ModifyFunctionSignature()
288 int64_t original_arg_count = region.getNumArguments(); in HandleWhileRegionOp() local
289 for (int64_t i = 0; i < original_arg_count; ++i) { in HandleWhileRegionOp()
|