Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc55 std::vector<const Edge*> edges_to_replace; in ReplaceEdgeSources() local
58 edges_to_replace.push_back(edge); in ReplaceEdgeSources()
61 for (auto edge : edges_to_replace) { in ReplaceEdgeSources()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc895 std::vector<const Edge*> edges_to_replace; in RewriteTensorListWithConstElement() local
898 edges_to_replace.push_back(e); in RewriteTensorListWithConstElement()
901 if (edges_to_replace.empty()) { in RewriteTensorListWithConstElement()
909 for (const Edge* e : edges_to_replace) { in RewriteTensorListWithConstElement()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc396 std::vector<const Edge*> edges_to_replace; in MoveHeadOutsideCompilationToHost() local
401 edges_to_replace.push_back(e); in MoveHeadOutsideCompilationToHost()
404 if (edges_to_replace.empty()) { in MoveHeadOutsideCompilationToHost()
409 for (const Edge* e : edges_to_replace) { in MoveHeadOutsideCompilationToHost()
2072 std::vector<const Edge*> edges_to_replace; in ReplaceOutputEdgesWithInputEdgeSourceForWhile() local
2075 edges_to_replace.push_back(e); in ReplaceOutputEdgesWithInputEdgeSourceForWhile()
2078 for (const Edge* e : edges_to_replace) { in ReplaceOutputEdgesWithInputEdgeSourceForWhile()