Searched refs:arg_to_input_edge_offset (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1906 const int arg_to_input_edge_offset, Graph* g, Node* n, in MakeIdentityNodesForArgsToLift() argument
1913 int input_edge_index = arg_index + arg_to_input_edge_offset; in MakeIdentityNodesForArgsToLift()
1998 const int arg_to_input_edge_offset, Graph* g, Node* n) { in CleanUpInEdges() argument
2001 if (i < arg_to_input_edge_offset) continue; in CleanUpInEdges()
2003 int arg_idx = i - arg_to_input_edge_offset; in CleanUpInEdges()
2017 index_mapping.at(arg_idx) + arg_to_input_edge_offset); in CleanUpInEdges()
|
| /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
508 g->AddEdge(outside_compilation_node, 0, n, i + arg_to_input_edge_offset); in AddEdgesFromOutsideCompilationNodes()
|