Method GetSoleRegst

oneflow/core/graph/task_node.cpp:418–423

}

std::shared_ptr<RegstDesc> TaskEdge::GetSoleRegst() const {
  CHECK_EQ(name_in_producer2regst_.size(), 1)
      << "edge: " << this << ", src: " << src_node()->task_id()
      << ", dst: " << dst_node()->task_id();
  return name_in_producer2regst_.begin()->second;
}

std::vector<std::shared_ptr<RegstDesc>> TaskEdge::GetRegsts() const {
  std::vector<std::shared_ptr<RegstDesc>> regst_descs;