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.cc466 const Edge* edge, const std::unordered_map<const Node*, Node*>& node_images, in RecordArg() argument
502 const std::unordered_map<const Node*, Node*>& node_images) { in RecordControlResult() argument
511 const std::unordered_map<const Node*, Node*>& node_images) { in RecordResult() argument
661 const std::unordered_map<const Node*, Node*>& node_images, in AddFunctionCallNode() argument
689 std::unordered_map<const Node*, Node*>* node_images) { in CopySubgraphNodes() argument
704 const std::unordered_map<const Node*, Node*>& node_images, in CopySubgraphEdges() argument
778 std::unordered_map<const Node*, Node*> node_images; in SplitIntoSubgraphs() local
820 Graph* graph_out, std::unordered_map<const Node*, Node*>* node_images) { in CopyNodesToOutputGraph() argument
837 const std::unordered_map<const Node*, Node*>& node_images, in AddFunctionCallNodes() argument
848 const std::unordered_map<const Node*, Node*>& node_images, in FindOutputImageOfEdgeSrc() argument
[all …]
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc3973 NodeToNodeReplicasMap* node_images, Graph* graph) { in CopyOutsideCompilationNodes()
4018 NodeToNodeReplicasMap* node_images, Graph* graph) { in ReplicateOutsideCompilationNodes()
4066 const NodeToNodeReplicasMap& node_images, in CopyOutsideCompilationEdges()
4194 const NodeToNodeReplicasMap& node_images, in ReplicateOutsideCompilationEdges()
4206 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