Home
last modified time | relevance | path

Searched refs:kGpuInferenceDevice (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_inference_session.h50 MS_REG_SESSION(kGpuInferenceDevice, GpuInferenceSession);
/third_party/mindspore/mindspore/ccsrc/cxx_api/graph/gpu/
Dgpu_graph_impl.cc78 session_impl_ = session::SessionFactory::Get().Create(kGpuInferenceDevice); in InitEnv()
80 … MS_LOG(ERROR) << "Session create failed!, please make sure target device:" << kGpuInferenceDevice in InitEnv()
/third_party/mindspore/mindspore/core/utils/
Dms_context.h55 const char kGpuInferenceDevice[] = "GpuInference"; variable