Searched refs:LabelSwitchTaskInfo (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | label_switch_task.h | 25 class LabelSwitchTask : public TaskRepeater<LabelSwitchTaskInfo> {
27 …LabelSwitchTask(const ModelContext &model_context, const std::shared_ptr<LabelSwitchTaskInfo> &tas…
36 std::shared_ptr<LabelSwitchTaskInfo> task_info_;
|
| D | label_switch_task.cc | 22 const std::shared_ptr<LabelSwitchTaskInfo> &task_info) in LabelSwitchTask()
23 : TaskRepeater<LabelSwitchTaskInfo>(model_context, task_info), in LabelSwitchTask()
76 REGISTER_TASK(TaskInfoType::LABEL_SWITCH, LabelSwitchTask, LabelSwitchTaskInfo);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | label_switch.cc | 22 using mindspore::ge::model_runner::LabelSwitchTaskInfo;
23 using LabelSwitchTaskInfoPtr = std::shared_ptr<LabelSwitchTaskInfo>;
69 …auto task_info_ptr = std::make_shared<LabelSwitchTaskInfo>(unique_name_, stream_id, label_size_, l… in GenTask()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 170 class LabelSwitchTaskInfo : public TaskInfo {
172 LabelSwitchTaskInfo(const std::string &op_name, uint32_t stream_id, uint32_t label_size, in LabelSwitchTaskInfo() function
178 ~LabelSwitchTaskInfo() override {} in ~LabelSwitchTaskInfo()
|
| /third_party/mindspore/tests/ut/cpp/device/ |
| D | ge_runtime_test.cc | 259 …std::make_shared<LabelSwitchTaskInfo>("op_name", 0, 2, std::vector<uint32_t>{0, 1}, reinterpret_ca… in TEST_F()
270 …std::make_shared<LabelSwitchTaskInfo>("op_name", 1, 2, std::vector<uint32_t>{0, 1}, reinterpret_ca… in TEST_F()
279 …std::make_shared<LabelSwitchTaskInfo>("op_name", 0, 3, std::vector<uint32_t>{0, 1, 2}, reinterpret… in TEST_F()
288 …std::make_shared<LabelSwitchTaskInfo>("op_name", 0, 2, std::vector<uint32_t>{0, 1}, reinterpret_ca… in TEST_F()
|