Searched defs: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()
761 Graph* xla_graph, Node* xla_node) { in RemoveUnusedXlaInput()
1000 const std::string& cluster_name, Graph* g, Graph* xla_graph, Node* xla_node, in MoveTailOutsideCompilationToHost()
1242 Graph* g, Graph* xla_graph, Node* xla_node) { in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1369 Graph* xla_graph, Node* xla_node) { in RemoveUnusedXlaOutput()
1480 Graph* xla_graph, Node* xla_node) { in RemoveEdgesBetweenArgAndRetval()
2904 Graph* xla_graph = xla_fbody->graph; in ProcessHeadTailOutsideCompilation() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 551 Graph *xla_graph = xla_fbody->graph; in TEST_F() local
822 Graph *xla_graph = xla_fbody->graph; in TEST_F() local
|