Home
last modified time | relevance | path

Searched defs:node_images (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc469 const Edge* edge, const std::unordered_map<const Node*, Node*>& node_images, in RecordArg() argument
505 const std::unordered_map<const Node*, Node*>& node_images) { in RecordControlResult() argument
514 const std::unordered_map<const Node*, Node*>& node_images) { in RecordResult() argument
664 const std::unordered_map<const Node*, Node*>& node_images, in AddFunctionCallNode() argument
692 std::unordered_map<const Node*, Node*>* node_images) { in CopySubgraphNodes() argument
707 const std::unordered_map<const Node*, Node*>& node_images, in CopySubgraphEdges() argument
781 std::unordered_map<const Node*, Node*> node_images; in SplitIntoSubgraphs() local
823 Graph* graph_out, std::unordered_map<const Node*, Node*>* node_images) { in CopyNodesToOutputGraph() argument
840 const std::unordered_map<const Node*, Node*>& node_images, in AddFunctionCallNodes() argument
851 const std::unordered_map<const Node*, Node*>& node_images, in FindOutputImageOfEdgeSrc() argument
[all …]
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc3466 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()
Dencapsulate_tpu_computations_pass.cc463 std::map<Node*, std::vector<Node*>> node_images; in MoveHeadOutsideCompilationToHost() local
1114 std::map<Node*, std::vector<Node*>> node_images; in MoveTailOutsideCompilationToHost() local