Home
last modified time | relevance | path

Searched defs:SetDevice (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/include/
Dconverter.h181 void Converter::SetDevice(const std::string &device) { SetDevice(StringToChar(device)); } in SetDevice() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/tensor/
Dtensor_impl.h141 void SetDevice(const std::string &device) override { in SetDevice() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/python/src/
Dtensor_numpy_impl.h115 void SetDevice(const std::string &device) override { device_ = device; } in SetDevice() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dtensor_default_impl.h109 void SetDevice(const std::string &device) override { device_ = device; } in SetDevice() function
Dtensor_utils.h101 void SetDevice(const std::string &device) override { in SetDevice() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/cxx_api/
Dconverter.cc300 void Converter::SetDevice(const std::vector<char> &device) { in SetDevice() function in mindspore::Converter
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dcuda_driver.cc305 bool CudaDriver::SetDevice(int index) { in SetDevice() function in mindspore::device::gpu::CudaDriver