Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtf2xla_util.cc1083 std::vector<const Edge*> edges_to_replace; in RewriteTensorListWithConstElement() local
1086 edges_to_replace.push_back(e); in RewriteTensorListWithConstElement()
1089 if (edges_to_replace.empty()) { in RewriteTensorListWithConstElement()
1096 for (const Edge* e : edges_to_replace) { in RewriteTensorListWithConstElement()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc370 std::vector<const Edge*> edges_to_replace; in MoveHeadOutsideCompilationToHost() local
375 edges_to_replace.push_back(e); in MoveHeadOutsideCompilationToHost()
378 if (edges_to_replace.empty()) { in MoveHeadOutsideCompilationToHost()
383 for (const Edge* e : edges_to_replace) { in MoveHeadOutsideCompilationToHost()
2026 std::vector<const Edge*> edges_to_replace; in ReplaceOutputEdgesWithInputEdgeSourceForWhile() local
2029 edges_to_replace.push_back(e); in ReplaceOutputEdgesWithInputEdgeSourceForWhile()
2032 for (const Edge* e : edges_to_replace) { in ReplaceOutputEdgesWithInputEdgeSourceForWhile()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc268 std::vector<const Edge*> edges_to_replace; in CreateInputProxy() local
272 edges_to_replace.push_back(input_edge); in CreateInputProxy()
291 for (const Edge* input_edge : edges_to_replace) { in CreateInputProxy()