Searched refs:LabelSetTask (Results 1 – 3 of 3) sorted by relevance
21 LabelSetTask::LabelSetTask(const ModelContext &model_context, const std::shared_ptr<LabelSetTaskInf… in LabelSetTask() function in mindspore::ge::model_runner::LabelSetTask40 LabelSetTask::~LabelSetTask() {} in ~LabelSetTask()42 void LabelSetTask::Distribute() { in Distribute()55 REGISTER_TASK(TaskInfoType::LABEL_SET, LabelSetTask, LabelSetTaskInfo);
24 class LabelSetTask : public TaskRepeater<LabelSetTaskInfo> {26 …LabelSetTask(const ModelContext &model_context, const std::shared_ptr<LabelSetTaskInfo> &task_info…28 ~LabelSetTask() override;
242 ASSERT_TRUE(std::dynamic_pointer_cast<LabelSetTask>(task) != nullptr); in TEST_F()