Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc363 std::vector<Node*> oc_nodes_at_head; in MoveHeadOutsideCompilationToHost() local
384 oc_nodes_at_head.push_back(n); in MoveHeadOutsideCompilationToHost()
389 for (Node* n : oc_nodes_at_head) { in MoveHeadOutsideCompilationToHost()
432 oc_nodes_at_head.erase( in MoveHeadOutsideCompilationToHost()
433 std::remove(oc_nodes_at_head.begin(), oc_nodes_at_head.end(), n), in MoveHeadOutsideCompilationToHost()
434 oc_nodes_at_head.end()); in MoveHeadOutsideCompilationToHost()
437 for (Node* n : oc_nodes_at_head) { in MoveHeadOutsideCompilationToHost()
464 for (Node* n : oc_nodes_at_head) { in MoveHeadOutsideCompilationToHost()
520 for (Node* n : oc_nodes_at_head) { in MoveHeadOutsideCompilationToHost()
687 for (Node* n : oc_nodes_at_head) { in MoveHeadOutsideCompilationToHost()
[all …]