Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc268 std::vector<Node*> call_nodes; in LowerFunctionalOps() local
271 call_nodes.push_back(n); in LowerFunctionalOps()
274 for (Node* call_node : call_nodes) { in LowerFunctionalOps()
278 if (!call_nodes.empty()) { in LowerFunctionalOps()
/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc516 std::vector<Node*> call_nodes; in RearrangeFunctionArguments() local
519 call_nodes.push_back(n); in RearrangeFunctionArguments()
522 for (Node* n : call_nodes) { in RearrangeFunctionArguments()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc940 std::vector<Node*> call_nodes; in PostprocessLiftedArgs() local
961 call_nodes.push_back(n); in PostprocessLiftedArgs()
965 for (Node* n : call_nodes) { in PostprocessLiftedArgs()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc2358 std::vector<Node*> while_nodes, if_nodes, call_nodes; in LiftOutsideCompilationOnlyArgs() local
2396 call_nodes.push_back(n); in LiftOutsideCompilationOnlyArgs()
2401 for (Node* call_node : call_nodes) { in LiftOutsideCompilationOnlyArgs()