MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / IsLbiOnTaskEdge

Function IsLbiOnTaskEdge

oneflow/core/graph/task_graph.cpp:202–207  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

200}
201
202bool IsLbiOnTaskEdge(const TaskEdge* edge, const LogicalBlobId& lbi) {
203 for (const auto& regst_desc : edge->GetRegsts()) {
204 if (regst_desc->HasLbi(lbi)) { return true; }
205 }
206 return false;
207}
208
209std::function<bool(const LogicalBlobId&, const std::string&)>
210MakePredicatorIsLbiAllConsumersReachable(

Calls 2

GetRegstsMethod · 0.80
HasLbiMethod · 0.80

Tested by

no test coverage detected