Searched refs:pivot_node (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1123 Node* pivot_node) { in ExpandHostGraphIntoMainGraph() argument
1146 if (pivot_node) { in ExpandHostGraphIntoMainGraph()
1147 node_map[host_graph->source_node()] = pivot_node; in ExpandHostGraphIntoMainGraph()
1203 Graph* host_graph, Node* pivot_node, in RewriteShapeInferenceGraph() argument
1250 Node* start_node = pivot_node ? pivot_node : host_graph->source_node(); in RewriteShapeInferenceGraph()
2530 Node* pivot_node = node_name_index[pivot_name]; in ExtractOutsideCompilation() local
2532 g, fld, host_graph_func_name, n, pivot_node)); in ExtractOutsideCompilation()
2538 shape_inference_graph_name, g, pivot_node, fld)); in ExtractOutsideCompilation()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 360 Node* pivot_node) { in MoveHeadOutsideCompilationToHost() argument
505 if (pivot_node) { in MoveHeadOutsideCompilationToHost()
506 g->AddControlEdge(pivot_node, copy_node); in MoveHeadOutsideCompilationToHost()
1001 Node* pivot_node) { in MoveTailOutsideCompilationToHost() argument
1163 if (pivot_node) { in MoveTailOutsideCompilationToHost()
1164 g->AddControlEdge(pivot_node, copy_node); in MoveTailOutsideCompilationToHost()
2896 Node* pivot_node = pivots[iter.first]; in ProcessHeadTailOutsideCompilation() local
2920 iter.second.cluster_name, g, xla_graph, iter.second.node, pivot_node)); in ProcessHeadTailOutsideCompilation()
2925 iter.second.cluster_name, g, xla_graph, iter.second.node, pivot_node)); in ProcessHeadTailOutsideCompilation()
|