Searched defs:outside_compilation_nodes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 318 std::map<string, DistributedTPURewritePass::OutsideCompilationNodeMap>* in FindTaggedNodes()
3463 int replica_index, const std::vector<Node*>& outside_compilation_nodes, in CopyOutsideCompilationNodes()
3510 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationNodes()
3558 const std::vector<Node*>& outside_compilation_nodes, in CopyOutsideCompilationEdges()
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() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 503 const std::vector<Node*>& outside_compilation_nodes, Graph* g, Node* n) { in AddEdgesFromOutsideCompilationNodes()
619 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForWhile() local
736 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForIf() local
902 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForCall() local
2372 std::vector<Node*> outside_compilation_nodes; in ExtractOutsideCompilationForFunction() local
|