Home
last modified time | relevance | path

Searched defs:model_context (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/tests/ut/cpp/device/
Dge_runtime_test.cc65 …ModelContext model_context(0, 0, 0, reinterpret_cast<rtModel_t>(1), reinterpret_cast<rtStream_t>(2… in TEST_F() local
72 …ModelContext model_context(0, 0, 0, reinterpret_cast<rtModel_t>(1), reinterpret_cast<rtStream_t>(2… in TEST_F() local
84 …ModelContext model_context(0, 0, 0, reinterpret_cast<rtModel_t>(1), reinterpret_cast<rtStream_t>(2… in TEST_F() local
96 …ModelContext model_context(0, 0, 0, reinterpret_cast<rtModel_t>(1), reinterpret_cast<rtStream_t>(2… in TEST_F() local
106 …ModelContext model_context(0, 0, 0, reinterpret_cast<rtModel_t>(1), reinterpret_cast<rtStream_t>(2… in TEST_F() local
116 …ModelContext model_context(0, 0, 0, reinterpret_cast<rtModel_t>(1), reinterpret_cast<rtStream_t>(2… in TEST_F() local
124 …ModelContext model_context(0, 0, 0, reinterpret_cast<rtModel_t>(1), reinterpret_cast<rtStream_t>(2… in TEST_F() local
134 …ModelContext model_context(0, 0, 0, reinterpret_cast<rtModel_t>(1), reinterpret_cast<rtStream_t>(2… in TEST_F() local
143 …ModelContext model_context(0, 0, 0, reinterpret_cast<rtModel_t>(1), reinterpret_cast<rtStream_t>(2… in TEST_F() local
162 ModelContext model_context(0, 0, 0, model, reinterpret_cast<rtStream_t>(2), {stream}, in TEST_F() local
[all …]
/third_party/mindspore/mindspore/lite/src/cxx_api/model/
Dmodel.cc32 … const std::shared_ptr<Context> &model_context, const Key &dec_key, const std::string &dec_mode) { in Build()
49 …(const std::string &model_path, ModelType model_type, const std::shared_ptr<Context> &model_contex… in Build()
67 Status Model::Build(GraphCell graph, const std::shared_ptr<Context> &model_context, in Build()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/
Dmemcpy_async_task.cc22 MemcpyAsyncTask::MemcpyAsyncTask(const ModelContext &model_context, in MemcpyAsyncTask()
Dtask.h51 TaskRepeater(const ModelContext &model_context, const std::shared_ptr<T> &task_info) {} in TaskRepeater()
Dstream_active_task.cc22 StreamActiveTask::StreamActiveTask(const ModelContext &model_context, in StreamActiveTask()
Devent_wait_task.cc22 EventWaitTask::EventWaitTask(const ModelContext &model_context, const std::shared_ptr<EventWaitTask… in EventWaitTask()
Devent_record_task.cc22 EventRecordTask::EventRecordTask(const ModelContext &model_context, in EventRecordTask()
Dlabel_set_task.cc21 LabelSetTask::LabelSetTask(const ModelContext &model_context, const std::shared_ptr<LabelSetTaskInf… in LabelSetTask()
Dtbe_task.cc24 TbeTask::TbeTask(const ModelContext &model_context, const std::shared_ptr<TbeTaskInfo> &task_info) in TbeTask()
Dlabel_switch_task.cc21 LabelSwitchTask::LabelSwitchTask(const ModelContext &model_context, in LabelSwitchTask()
Dstream_switch_task.cc23 StreamSwitchTask::StreamSwitchTask(const ModelContext &model_context, in StreamSwitchTask()
Dprofiler_task.cc21 ProfilerTask::ProfilerTask(const ModelContext &model_context, const std::shared_ptr<ProfilerTraceTa… in ProfilerTask()
Dlabel_goto_task.cc22 LabelGotoTask::LabelGotoTask(const ModelContext &model_context, const std::shared_ptr<LabelGotoTask… in LabelGotoTask()
Dtask_factory.h51 …std::shared_ptr<Task> Create(const ModelContext &model_context, const std::shared_ptr<TaskInfo> &t… in Create()
Daicpu_task.cc26 AicpuTask::AicpuTask(const ModelContext &model_context, const std::shared_ptr<AicpuTaskInfo> &task_… in AicpuTask()
Dhccl_task.cc28 HcclTask::HcclTask(const ModelContext &model_context, const std::shared_ptr<HcclTaskInfo> &task_inf… in HcclTask()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/
Dmodel_impl.h71 void SetContext(const std::shared_ptr<Context> &model_context) { in SetContext()
Dmodel.cc36 Status Model::Build(GraphCell graph_cell, const std::shared_ptr<Context> &model_context, in Build()
/third_party/mindspore/mindspore/lite/src/c_api/
Dmodel_c.cc56 …ld(const void *model_data, size_t data_size, ModelType model_type, const ContextC *model_context) { in Build()
77 …ModelC::Build(const std::string &model_path, ModelType model_type, const ContextC *model_context) { in Build()
314 OH_AI_ModelType model_type, const OH_AI_ContextHandle model_context) { in OH_AI_ModelBuild()
330 const OH_AI_ContextHandle model_context) { in OH_AI_ModelBuildFromFile()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/
Druntime_model.cc158 …ModelContext model_context(device_id, session_id, davinci_model->GetPriority(), rt_model_handle_, … in GenerateTask() local
/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.cc228 auto model_context = std::make_shared<mindspore::Context>(); in SetAclModelOptions() local