Searched refs:NodeToNodeReplicasMap (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.h | 469 typedef absl::node_hash_map<Node*, std::vector<Node*>> NodeToNodeReplicasMap; typedef
485 NodeToNodeReplicasMap* node_images, Graph* graph);
493 NodeToNodeReplicasMap* node_images, Graph* graph);
499 const NodeToNodeReplicasMap& node_images,
507 const NodeToNodeReplicasMap& node_images,
514 const NodeToNodeReplicasMap& node_images, Graph* graph);
569 NodeToNodeReplicasMap* outside_compilation_node_images, Graph* graph,
|
| H A D | distributed_tpu_rewrite_pass.cc | 3922 NodeToNodeReplicasMap* node_images, Graph* graph) { in CopyOutsideCompilationNodes()
3965 NodeToNodeReplicasMap* node_images, Graph* graph) { in ReplicateOutsideCompilationNodes()
4013 const NodeToNodeReplicasMap& node_images, in CopyOutsideCompilationEdges()
4141 const NodeToNodeReplicasMap& node_images, in ReplicateOutsideCompilationEdges()
4153 const NodeToNodeReplicasMap& node_images, Graph* graph) { in RemoveOutsideCompilationNodes()
4716 NodeToNodeReplicasMap* outside_compilation_node_images, Graph* graph, in RewriteTPUReplicateNode()
5003 NodeToNodeReplicasMap outside_compilation_node_images; in InternalRun()
|