Searched refs:arg_to_input_edge_offset (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | encapsulate_tpu_computations_pass.cc | 1864 const int arg_to_input_edge_offset, Graph* g, Node* n, in MakeIdentityNodesForArgsToLift() argument
1871 int input_edge_index = arg_index + arg_to_input_edge_offset; in MakeIdentityNodesForArgsToLift()
1952 const int arg_to_input_edge_offset, Graph* g, Node* n) { in CleanUpInEdges() argument
1955 if (i < arg_to_input_edge_offset) continue; in CleanUpInEdges()
1957 int arg_idx = i - arg_to_input_edge_offset; in CleanUpInEdges()
1971 index_mapping.at(arg_idx) + arg_to_input_edge_offset); in CleanUpInEdges()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass.cc | 501 const int original_arg_count, const int arg_to_input_edge_offset, in AddEdgesFromOutsideCompilationNodes() argument
508 g->AddEdge(outside_compilation_node, 0, n, i + arg_to_input_edge_offset); in AddEdgesFromOutsideCompilationNodes()
|