Searched refs:OutsideCompilationNodeMap (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.h | 477 typedef std::map<string, std::vector<Node*>> OutsideCompilationNodeMap; typedef
492 const OutsideCompilationNodeMap& outside_compilation_nodes,
506 const OutsideCompilationNodeMap& outside_compilation_nodes,
528 const OutsideCompilationNodeMap& outside_compilation_nodes,
567 const OutsideCompilationNodeMap& outside_compilation_nodes,
|
| H A D | distributed_tpu_rewrite_pass.cc | 334 std::map<string, DistributedTPURewritePass::OutsideCompilationNodeMap>* in FindTaggedNodes()
359 DistributedTPURewritePass::OutsideCompilationNodeMap(); in FindTaggedNodes()
3964 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationNodes()
4140 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationEdges()
4354 const OutsideCompilationNodeMap& outside_compilation_nodes, in ParseHostComputeCores()
4714 const OutsideCompilationNodeMap& outside_compilation_nodes, in RewriteTPUReplicateNode()
4965 std::map<string, OutsideCompilationNodeMap> outside_compilation_nodes; in InternalRun()
|