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

Method BindEdgeWithProducedRegst

oneflow/core/graph/task_node.cpp:300–303  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

298}
299
300void TaskNode::BindEdgeWithProducedRegst(TaskEdge* edge, const std::string& name) {
301 if (edge->HasRegst(name)) { return; }
302 edge->AddRegst(name, GetProducedRegst(name));
303}
304
305std::shared_ptr<RegstDesc> TaskNode::GetAndCheckRegst(const std::string& name,
306 bool enable_reuse_mem,

Callers 6

ConnectCtrlEdgeMethod · 0.80
FOR_RANGEFunction · 0.80
FOR_RANGEFunction · 0.80
FOR_RANGEMethod · 0.80
FOR_RANGEFunction · 0.80

Calls 2

AddRegstMethod · 0.80
HasRegstMethod · 0.45

Tested by

no test coverage detected