Method HasRegst

oneflow/core/graph/task_node.cpp:414–416

}

bool TaskEdge::HasRegst(const std::string& name_in_producer) const {
  return (name_in_producer2regst_.find(name_in_producer) != name_in_producer2regst_.end());
}

std::shared_ptr<RegstDesc> TaskEdge::GetSoleRegst() const {
  CHECK_EQ(name_in_producer2regst_.size(), 1)