Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dencapsulate_subgraphs_pass.cc450 const Edge* edge, const std::unordered_map<const Node*, Node*>& node_images, in RecordArg() argument
484 const std::unordered_map<const Node*, Node*>& node_images) { in RecordControlResult() argument
493 const std::unordered_map<const Node*, Node*>& node_images) { in RecordResult() argument
640 const std::unordered_map<const Node*, Node*>& node_images, in AddFunctionCallNode() argument
666 std::unordered_map<const Node*, Node*>* node_images) { in CopySubgraphNodes() argument
681 const std::unordered_map<const Node*, Node*>& node_images, in CopySubgraphEdges() argument
755 std::unordered_map<const Node*, Node*> node_images; in SplitIntoSubgraphs() local
797 Graph* graph_out, std::unordered_map<const Node*, Node*>* node_images) { in CopyNodesToOutputGraph() argument
814 const std::unordered_map<const Node*, Node*>& node_images, in AddFunctionCallNodes() argument
825 const std::unordered_map<const Node*, Node*>& node_images, in FindOutputImageOfEdgeSrc() argument
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc3922 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()
H A Dencapsulate_tpu_computations_pass.cc437 std::map<Node*, std::vector<Node*>> node_images; in MoveHeadOutsideCompilationToHost() local
1083 std::map<Node*, std::vector<Node*>> node_images; in MoveTailOutsideCompilationToHost() local