Searched refs:TbeTaskInfo (Results 1 – 6 of 6) sorted by relevance
25 class TbeTask : public TaskRepeater<TbeTaskInfo> {27 TbeTask(const ModelContext &model_context, const std::shared_ptr<TbeTaskInfo> &task_info);38 std::shared_ptr<TbeTaskInfo> task_info_;
24 TbeTask::TbeTask(const ModelContext &model_context, const std::shared_ptr<TbeTaskInfo> &task_info) in TbeTask()25 : TaskRepeater<TbeTaskInfo>(model_context, task_info), in TbeTask()96 REGISTER_TASK(TaskInfoType::TBE, TbeTask, TbeTaskInfo);
32 using TbeTaskInfoPtr = std::shared_ptr<mindspore::ge::model_runner::TbeTaskInfo>;132 TbeTaskInfoPtr task_info_ptr = std::make_shared<mindspore::ge::model_runner::TbeTaskInfo>( in GenTask()
26 using TbeTaskInfoPtr = std::shared_ptr<mindspore::ge::model_runner::TbeTaskInfo>;107 TbeTaskInfoPtr task_info_ptr = std::make_shared<mindspore::ge::model_runner::TbeTaskInfo>( in GenTask()
66 class TbeTaskInfo : public TaskInfo {68 …TbeTaskInfo(const std::string &op_name, uint32_t stream_id, const std::string &stub_func, uint32_t… in TbeTaskInfo() function84 ~TbeTaskInfo() override {} in ~TbeTaskInfo()
397 std::shared_ptr<TaskInfo> tbe_task_info = std::make_shared<TbeTaskInfo>( in TEST_F()415 std::shared_ptr<TaskInfo> tbe_task_info = std::make_shared<TbeTaskInfo>( in TEST_F()429 std::shared_ptr<TaskInfo> tbe_task_info = std::make_shared<TbeTaskInfo>( in TEST_F()445 std::shared_ptr<TaskInfo> tbe_task_info = std::make_shared<TbeTaskInfo>( in TEST_F()