MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / UpdateConsumers

Method UpdateConsumers

tensorflow/core/common_runtime/lower_while_op.cc:394–406  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

392}
393
394Status LowerWhileHelper::UpdateConsumers() {
395 for (const Edge* e : while_op_->out_edges()) {
396 if (e->IsControlEdge()) {
397 graph_->AddControlEdge(lowered_while_executed_, e->dst());
398 } else {
399 // Feed the outputs directly from the exit nodes so that downstream ops
400 // can start before all the outputs have been computed.
401 graph_->AddEdge(exit_nodes_[e->src_output()], 0, e->dst(),
402 e->dst_input());
403 }
404 }
405 return Status::OK();
406}
407
408string LowerWhileHelper::NewName(const string& infix) {
409 return graph_->NewName(strings::StrCat(name_, "/", infix));

Callers

nothing calls this directly

Calls 6

IsControlEdgeMethod · 0.45
AddControlEdgeMethod · 0.45
dstMethod · 0.45
AddEdgeMethod · 0.45
src_outputMethod · 0.45
dst_inputMethod · 0.45

Tested by

no test coverage detected