Searched refs:LabelSetTaskInfo (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | label_set.cc | 22 using mindspore::ge::model_runner::LabelSetTaskInfo;
23 using LabelSetTaskInfoPtr = std::shared_ptr<LabelSetTaskInfo>;
55 …std::shared_ptr<LabelSetTaskInfo> task_info_ptr = std::make_shared<LabelSetTaskInfo>(unique_name_,… in GenTask()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | label_set_task.h | 24 class LabelSetTask : public TaskRepeater<LabelSetTaskInfo> {
26 …LabelSetTask(const ModelContext &model_context, const std::shared_ptr<LabelSetTaskInfo> &task_info…
33 std::shared_ptr<LabelSetTaskInfo> task_info_;
|
| D | label_set_task.cc | 21 …LabelSetTask(const ModelContext &model_context, const std::shared_ptr<LabelSetTaskInfo> &task_info) in LabelSetTask()
22 : TaskRepeater<LabelSetTaskInfo>(model_context, task_info), in LabelSetTask()
55 REGISTER_TASK(TaskInfoType::LABEL_SET, LabelSetTask, LabelSetTaskInfo);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 148 class LabelSetTaskInfo : public TaskInfo {
150 LabelSetTaskInfo(const std::string &op_name, uint32_t stream_id, uint32_t label_id) in LabelSetTaskInfo() function
152 ~LabelSetTaskInfo() override {} in ~LabelSetTaskInfo()
|
| D | runtime_model.cc | 114 auto label_set_task_info = std::static_pointer_cast<LabelSetTaskInfo>(task_info); in InitLabel()
|
| /third_party/mindspore/tests/ut/cpp/device/ |
| D | ge_runtime_test.cc | 240 …std::shared_ptr<TaskInfo> label_set_task_info = std::make_shared<LabelSetTaskInfo>("op_name", 0, 0… in TEST_F()
|