Home
last modified time | relevance | path

Searched refs:outside_compilation_nodes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h470 int replica_index, const std::vector<Node*>& outside_compilation_nodes,
480 const OutsideCompilationNodeMap& outside_compilation_nodes,
486 const std::vector<Node*>& outside_compilation_nodes,
494 const OutsideCompilationNodeMap& outside_compilation_nodes,
516 const OutsideCompilationNodeMap& outside_compilation_nodes,
554 const OutsideCompilationNodeMap& outside_compilation_nodes,
Ddistributed_tpu_rewrite_pass.cc319 outside_compilation_nodes, in FindTaggedNodes()
334 if (outside_compilation_nodes->find(cluster) != in FindTaggedNodes()
335 outside_compilation_nodes->end()) { in FindTaggedNodes()
342 (*outside_compilation_nodes)[cluster] = in FindTaggedNodes()
365 const auto iter = outside_compilation_nodes->find(cluster); in FindTaggedNodes()
366 if (iter == outside_compilation_nodes->end()) { in FindTaggedNodes()
3463 int replica_index, const std::vector<Node*>& outside_compilation_nodes, in CopyOutsideCompilationNodes() argument
3467 for (Node* node : outside_compilation_nodes) { in CopyOutsideCompilationNodes()
3510 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationNodes() argument
3515 for (const auto& oc_cluster_iter : outside_compilation_nodes) { in ReplicateOutsideCompilationNodes()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc503 const std::vector<Node*>& outside_compilation_nodes, Graph* g, Node* n) { in AddEdgesFromOutsideCompilationNodes() argument
507 outside_compilation_nodes[i - original_arg_count]; in AddEdgesFromOutsideCompilationNodes()
619 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForWhile() local
623 std::back_inserter(outside_compilation_nodes), in PostprocessLiftedArgsForWhile()
627 data_types, outside_compilation_nodes, g, in PostprocessLiftedArgsForWhile()
736 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForIf() local
740 outside_compilation_nodes.push_back(pair.second); in PostprocessLiftedArgsForIf()
745 if (std::find(outside_compilation_nodes.begin(), in PostprocessLiftedArgsForIf()
746 outside_compilation_nodes.end(), in PostprocessLiftedArgsForIf()
747 pair.second) == outside_compilation_nodes.end()) { in PostprocessLiftedArgsForIf()
[all …]