Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc3585 const auto& src_images = iter->second; in CopyOutsideCompilationEdges() local
3586 if (src_images.size() != images.size()) { in CopyOutsideCompilationEdges()
3589 " in an outside_compilation block replicated ", src_images.size(), in CopyOutsideCompilationEdges()
3607 if (src_images.size() == 1) { in CopyOutsideCompilationEdges()
3610 for (int i = 0; i < src_images.size(); ++i) { in CopyOutsideCompilationEdges()
3629 if (src_images.size() == 1) { in CopyOutsideCompilationEdges()
3632 for (int i = 0; i < src_images.size(); ++i) { in CopyOutsideCompilationEdges()
3641 for (int i = 0; i < src_images.size(); ++i) { in CopyOutsideCompilationEdges()
3642 graph->AddEdge(src_images[i], edge->src_output(), images[i], in CopyOutsideCompilationEdges()