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

Method GetRegsts

oneflow/core/graph/task_node.cpp:425–430  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

423}
424
425std::vector<std::shared_ptr<RegstDesc>> TaskEdge::GetRegsts() const {
426 std::vector<std::shared_ptr<RegstDesc>> regst_descs;
427 regst_descs.reserve(name_in_producer2regst_.size());
428 for (auto& pair : name_in_producer2regst_) { regst_descs.emplace_back(pair.second); }
429 return regst_descs;
430}
431
432void TaskEdge::AddRegst(const std::string& name_in_producer,
433 const std::shared_ptr<RegstDesc>& regst) {

Callers 4

ConsumeAllRegstsMethod · 0.80
ForEachDataEdgeFunction · 0.80
IsLbiOnTaskEdgeFunction · 0.80

Calls 2

sizeMethod · 0.45
emplace_backMethod · 0.45

Tested by

no test coverage detected