Searched refs:davinci_model (Results 1 – 4 of 4) sorted by relevance
51 void RuntimeModel::InitStream(const std::shared_ptr<DavinciModel> &davinci_model) { in InitStream() argument52 MS_EXCEPTION_IF_NULL(davinci_model); in InitStream()57 for (const auto &stream_id : davinci_model->GetWaitActiveStreams()) { in InitStream()62 for (const auto &stream_id : davinci_model->GetForceCopyStreams()) { in InitStream()67 MS_LOG(INFO) << "Total stream num " << davinci_model->GetStreamNum(); in InitStream()68 for (uint32_t i = 0; i < davinci_model->GetStreamNum(); ++i) { in InitStream()74 rtError_t rt_ret = rtStreamCreateWithFlags(&stream, davinci_model->GetPriority(), flag); in InitStream()105 void RuntimeModel::InitLabel(const std::shared_ptr<DavinciModel> &davinci_model) { in InitLabel() argument106 MS_LOG(INFO) << "Label number: " << davinci_model->GetBatchNum(); in InitLabel()107 label_list_.resize(davinci_model->GetBatchNum()); in InitLabel()[all …]
36 … Load(uint32_t device_id, uint64_t session_id, const std::shared_ptr<DavinciModel> &davinci_model);47 void InitResource(const std::shared_ptr<DavinciModel> &davinci_model);48 …eTask(uint32_t device_id, uint64_t session_id, const std::shared_ptr<DavinciModel> &davinci_model);49 void InitStream(const std::shared_ptr<DavinciModel> &davinci_model);51 void InitLabel(const std::shared_ptr<DavinciModel> &davinci_model);
29 const std::shared_ptr<DavinciModel> &davinci_model) { in LoadDavinciModel() argument31 model->Load(device_id, session_id, davinci_model); in LoadDavinciModel()
35 const std::shared_ptr<DavinciModel> &davinci_model);