Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc253 std::vector<Node*> if_nodes; in LowerFunctionalOps() local
256 if_nodes.push_back(n); in LowerFunctionalOps()
259 for (Node* if_node : if_nodes) { in LowerFunctionalOps()
263 if (!if_nodes.empty()) { in LowerFunctionalOps()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc2118 std::vector<Node*> if_nodes, while_nodes, func_call_nodes; in ExtractOutsideCompilationForNodesWithAssociatedFunctions() local
2121 if_nodes.push_back(n); in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
2136 for (Node* n : if_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
2393 if_nodes.push_back(n); in LiftOutsideCompilationOnlyArgs()
2480 for (Node* n : if_nodes) { in LiftOutsideCompilationOnlyArgs()