Searched refs:StreamSwitchTaskInfo (Results 1 – 5 of 5) sorted by relevance
25 class StreamSwitchTask : public TaskRepeater<StreamSwitchTaskInfo> {27 …StreamSwitchTask(const ModelContext &model_context, const std::shared_ptr<StreamSwitchTaskInfo> &t…34 std::shared_ptr<StreamSwitchTaskInfo> task_info_;
24 const std::shared_ptr<StreamSwitchTaskInfo> &task_info) in StreamSwitchTask()25 : TaskRepeater<StreamSwitchTaskInfo>(model_context, task_info), in StreamSwitchTask()70 REGISTER_TASK(TaskInfoType::STREAM_SWITCH, StreamSwitchTask, StreamSwitchTaskInfo);
24 using mindspore::ge::model_runner::StreamSwitchTaskInfo;25 using StreamSwitchTaskInfoPtr = std::shared_ptr<StreamSwitchTaskInfo>;101 std::shared_ptr<StreamSwitchTaskInfo> task_info_ptr = std::make_shared<StreamSwitchTaskInfo>( in GenTask()
326 class StreamSwitchTaskInfo : public TaskInfo {328 …StreamSwitchTaskInfo(const std::string &op_name, uint32_t stream_id, int64_t true_stream_id, void … in StreamSwitchTaskInfo() function336 ~StreamSwitchTaskInfo() override {} in ~StreamSwitchTaskInfo()
363 std::shared_ptr<TaskInfo> stream_switch_task_info = std::make_shared<StreamSwitchTaskInfo>( in TEST_F()375 std::shared_ptr<TaskInfo> stream_switch_task_info = std::make_shared<StreamSwitchTaskInfo>( in TEST_F()387 std::shared_ptr<TaskInfo> stream_switch_task_info = std::make_shared<StreamSwitchTaskInfo>( in TEST_F()