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