Searched defs:xla_node (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | encapsulate_tpu_computations_pass.cc | 307 Node* xla_node) { in UpdateMirroredVariableIndices()
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()
|