Searched refs:NodeToNodeReplicasMap (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 457 typedef absl::node_hash_map<Node*, std::vector<Node*>> NodeToNodeReplicasMap; typedef
473 NodeToNodeReplicasMap* node_images, Graph* graph);
481 NodeToNodeReplicasMap* node_images, Graph* graph);
487 const NodeToNodeReplicasMap& node_images,
495 const NodeToNodeReplicasMap& node_images,
502 const NodeToNodeReplicasMap& node_images, Graph* graph);
556 NodeToNodeReplicasMap* outside_compilation_node_images, Graph* graph,
|
| D | distributed_tpu_rewrite_pass.cc | 3466 NodeToNodeReplicasMap* node_images, Graph* graph) { in CopyOutsideCompilationNodes()
3511 NodeToNodeReplicasMap* node_images, Graph* graph) { in ReplicateOutsideCompilationNodes()
3559 const NodeToNodeReplicasMap& node_images, in CopyOutsideCompilationEdges()
3687 const NodeToNodeReplicasMap& node_images, in ReplicateOutsideCompilationEdges()
3699 const NodeToNodeReplicasMap& node_images, Graph* graph) { in RemoveOutsideCompilationNodes()
4215 NodeToNodeReplicasMap* outside_compilation_node_images, Graph* graph, in RewriteTPUReplicateNode()
4477 NodeToNodeReplicasMap outside_compilation_node_images; in Run()
|