Searched refs:label_list_ (Results 1 – 6 of 6) sorted by relevance
28 label_list_ = {}; in LabelSwitchKernel()44 label_list_ = GetValue<std::vector<uint32_t>>(primitive->GetAttr(kAttrLabelSwitchList)); in Init()45 label_size_ = SizeToUint(label_list_.size()); in Init()47 for (auto label : label_list_) { in Init()69 … = std::make_shared<LabelSwitchTaskInfo>(unique_name_, stream_id, label_size_, label_list_, cond_); in GenTask()
40 std::vector<uint32_t> label_list_;
35 label_list_(label_list), in ModelContext()45 const std::vector<rtLabel_t> &label_list() const { return label_list_; } in label_list()55 std::vector<rtLabel_t> label_list_; variable
107 label_list_.resize(davinci_model->GetBatchNum()); in InitLabel()126 label_list_[label_set_task_info->label_id()] = rt_label; in InitLabel()159 stream_list_, label_list_, event_list_); in GenerateTask()263 for (size_t i = 0; i < label_list_.size(); i++) { in RtLabelDestory()264 if (label_list_[i] == nullptr) { in RtLabelDestory()267 if (rtLabelDestroy(label_list_[i]) != RT_ERROR_NONE) { in RtLabelDestory()
176 label_list_(label_list), in LabelSwitchTaskInfo()180 const std::vector<uint32_t> &label_list() const { return label_list_; } in label_list()185 std::vector<uint32_t> label_list_; variable
62 std::vector<rtLabel_t> label_list_{};