Searched refs:ProfilerTask (Results 1 – 3 of 3) sorted by relevance
21 ProfilerTask::ProfilerTask(const ModelContext &model_context, const std::shared_ptr<ProfilerTraceTa… in ProfilerTask() function in mindspore::ge::model_runner::ProfilerTask33 ProfilerTask::~ProfilerTask() {} in ~ProfilerTask()35 void ProfilerTask::Distribute() { in Distribute()57 REGISTER_TASK(TaskInfoType::PROFILER_TRACE, ProfilerTask, ProfilerTraceTaskInfo);
24 class ProfilerTask : public TaskRepeater<ProfilerTraceTaskInfo> {26 …ProfilerTask(const ModelContext &model_context, const std::shared_ptr<ProfilerTraceTaskInfo> &task…28 ~ProfilerTask() override;
326 ASSERT_TRUE(std::dynamic_pointer_cast<ProfilerTask>(task) != nullptr); in TEST_F()