Searched defs:xla_graph (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 | 333 const std::string& cluster_name, Graph* g, Graph* xla_graph, Node* xla_node, in MoveHeadOutsideCompilationToHost()
730 Graph* xla_graph, Node* xla_node) { in RemoveUnusedXlaInput()
969 const std::string& cluster_name, Graph* g, Graph* xla_graph, Node* xla_node, in MoveTailOutsideCompilationToHost()
1205 Graph* g, Graph* xla_graph, Node* xla_node) { in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1328 Graph* xla_graph, Node* xla_node) { in RemoveUnusedXlaOutput()
1439 Graph* xla_graph, Node* xla_node) { in RemoveEdgesBetweenArgAndRetval()
2852 Graph* xla_graph = xla_fbody->graph; in ProcessHeadTailOutsideCompilation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A 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
|