Searched refs:OutsideCompilationNodeMap (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 465 typedef std::map<string, std::vector<Node*>> OutsideCompilationNodeMap; typedef
480 const OutsideCompilationNodeMap& outside_compilation_nodes,
494 const OutsideCompilationNodeMap& outside_compilation_nodes,
516 const OutsideCompilationNodeMap& outside_compilation_nodes,
554 const OutsideCompilationNodeMap& outside_compilation_nodes,
|
| D | distributed_tpu_rewrite_pass.cc | 318 std::map<string, DistributedTPURewritePass::OutsideCompilationNodeMap>* in FindTaggedNodes()
343 DistributedTPURewritePass::OutsideCompilationNodeMap(); in FindTaggedNodes()
3510 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationNodes()
3686 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationEdges()
3900 const OutsideCompilationNodeMap& outside_compilation_nodes, in ParseHostComputeCores()
4213 const OutsideCompilationNodeMap& outside_compilation_nodes, in RewriteTPUReplicateNode()
4452 std::map<string, OutsideCompilationNodeMap> outside_compilation_nodes; in Run()
|