Searched refs:LabelGotoTaskInfo (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | label_goto.cc | 22 using mindspore::ge::model_runner::LabelGotoTaskInfo;
23 using LabelGotoTaskInfoPtr = std::shared_ptr<LabelGotoTaskInfo>;
55 std::shared_ptr<LabelGotoTaskInfo> task_info_ptr = in GenTask()
56 std::make_shared<LabelGotoTaskInfo>(unique_name_, stream_id, label_); in GenTask()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | label_goto_task.h | 28 class LabelGotoTask : public TaskRepeater<LabelGotoTaskInfo> {
30 …LabelGotoTask(const ModelContext &model_context, const std::shared_ptr<LabelGotoTaskInfo> &task_in…
37 std::shared_ptr<LabelGotoTaskInfo> task_info_;
|
| D | label_goto_task.cc | 22 …belGotoTask(const ModelContext &model_context, const std::shared_ptr<LabelGotoTaskInfo> &task_info) in LabelGotoTask()
23 : TaskRepeater<LabelGotoTaskInfo>(model_context, task_info), in LabelGotoTask()
82 REGISTER_TASK(TaskInfoType::LABEL_GOTO, LabelGotoTask, LabelGotoTaskInfo);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 159 class LabelGotoTaskInfo : public TaskInfo {
161 LabelGotoTaskInfo(const std::string &op_name, uint32_t stream_id, uint32_t label_id) in LabelGotoTaskInfo() function
163 ~LabelGotoTaskInfo() override {} in ~LabelGotoTaskInfo()
|
| /third_party/mindspore/tests/ut/cpp/device/ |
| D | ge_runtime_test.cc | 204 …std::shared_ptr<TaskInfo> label_goto_task_info = std::make_shared<LabelGotoTaskInfo>("op_name", 0,… in TEST_F()
216 …std::shared_ptr<TaskInfo> label_goto_task_info = std::make_shared<LabelGotoTaskInfo>("op_name", 0,… in TEST_F()
224 …std::shared_ptr<TaskInfo> label_goto_task_info = std::make_shared<LabelGotoTaskInfo>("op_name", 0,… in TEST_F()
250 …std::shared_ptr<TaskInfo> label_set_task_info = std::make_shared<LabelGotoTaskInfo>("op_name", 0, … in TEST_F()
|