Home
last modified time | relevance | path

Searched refs:node_images (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc210 const std::unordered_map<const Node*, Node*>& node_images,
228 const std::unordered_map<const Node*, Node*>& node_images,
236 const std::unordered_map<const Node*, Node*>& node_images);
243 const std::unordered_map<const Node*, Node*>& node_images);
307 const std::unordered_map<const Node*, Node*>& node_images,
311 Status CopySubgraphNodes(std::unordered_map<const Node*, Node*>* node_images);
315 Graph* graph_out, std::unordered_map<const Node*, Node*>* node_images);
319 const std::unordered_map<const Node*, Node*>& node_images,
327 const std::unordered_map<const Node*, Node*>& node_images,
342 const std::unordered_map<const Node*, Node*>& node_images,
[all …]
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h473 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);
Ddistributed_tpu_rewrite_pass.cc3466 NodeToNodeReplicasMap* node_images, Graph* graph) { in CopyOutsideCompilationNodes() argument
3500 std::vector<Node*>& node_image_vector = (*node_images)[node]; in CopyOutsideCompilationNodes()
3511 NodeToNodeReplicasMap* node_images, Graph* graph) { in ReplicateOutsideCompilationNodes() argument
3538 (*node_images)[node] = {node}; in ReplicateOutsideCompilationNodes()
3550 partial_device, node_images, graph)); in ReplicateOutsideCompilationNodes()
3559 const NodeToNodeReplicasMap& node_images, in CopyOutsideCompilationEdges() argument
3563 const auto& images = node_images.at(node); in CopyOutsideCompilationEdges()
3570 const auto iter = node_images.find(src); in CopyOutsideCompilationEdges()
3571 if (iter == node_images.end()) { in CopyOutsideCompilationEdges()
3649 const auto iter = node_images.find(dst); in CopyOutsideCompilationEdges()
[all …]
Dencapsulate_tpu_computations_pass.cc463 std::map<Node*, std::vector<Node*>> node_images; in MoveHeadOutsideCompilationToHost() local
494 g->AddEdge(node_images[e->src()][replica_id], e->src_output(), in MoveHeadOutsideCompilationToHost()
509 node_images[n].push_back(copy_node); in MoveHeadOutsideCompilationToHost()
523 node_images.find(e->dst()) == node_images.end()) { in MoveHeadOutsideCompilationToHost()
640 Node* src = node_images[original_src][replica_id]; in MoveHeadOutsideCompilationToHost()
1114 std::map<Node*, std::vector<Node*>> node_images; in MoveTailOutsideCompilationToHost() local
1149 node_images[e->dst()][replica_id], e->dst_input()); in MoveTailOutsideCompilationToHost()
1167 node_images[n].push_back(copy_node); in MoveTailOutsideCompilationToHost()
1177 Node* dst = node_images[original_edge->dst()][replica_id]; in MoveTailOutsideCompilationToHost()