Searched refs:xla_graph (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 359 const std::string& cluster_name, Graph* g, Graph* xla_graph, Node* xla_node, in MoveHeadOutsideCompilationToHost() argument
366 *xla_graph, /*enter=*/nullptr, in MoveHeadOutsideCompilationToHost()
370 if (e->src() == xla_graph->source_node()) { in MoveHeadOutsideCompilationToHost()
408 Node* const_copy = xla_graph->CopyNode(n); in MoveHeadOutsideCompilationToHost()
412 xla_graph->RemoveEdge(e); in MoveHeadOutsideCompilationToHost()
413 xla_graph->AddEdge(const_copy, 0, dst, dst_input); in MoveHeadOutsideCompilationToHost()
416 xla_graph->AddControlEdge(xla_graph->source_node(), const_copy); in MoveHeadOutsideCompilationToHost()
431 xla_graph->RemoveNode(n); in MoveHeadOutsideCompilationToHost()
478 if (e->src() == xla_graph->source_node()) { in MoveHeadOutsideCompilationToHost()
649 for (Node* n : xla_graph->nodes()) { in MoveHeadOutsideCompilationToHost()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 551 Graph *xla_graph = xla_fbody->graph; in TEST_F() local
552 auto node_name_index = xla_graph->BuildNodeNameIndex(); in TEST_F()
822 Graph *xla_graph = xla_fbody->graph; in TEST_F() local
823 auto node_name_index = xla_graph->BuildNodeNameIndex(); in TEST_F()
|