Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc2118 std::vector<Node*> if_nodes, while_nodes, func_call_nodes; in ExtractOutsideCompilationForNodesWithAssociatedFunctions() local
2123 while_nodes.push_back(n); in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
2143 for (Node* n : while_nodes) { in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc2358 std::vector<Node*> while_nodes, if_nodes, call_nodes; in LiftOutsideCompilationOnlyArgs() local
2373 while_nodes.push_back(n); in LiftOutsideCompilationOnlyArgs()
2473 for (Node* n : while_nodes) { in LiftOutsideCompilationOnlyArgs()