Home
last modified time | relevance | path

Searched defs:SetDeviceID (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/
Dcontext.cc429 void GPUDeviceInfo::SetDeviceID(uint32_t device_id) { in SetDeviceID() function in mindspore::GPUDeviceInfo
472 void AscendDeviceInfo::SetDeviceID(uint32_t device_id) { in SetDeviceID() function in mindspore::AscendDeviceInfo
693 void NNRTDeviceInfo::SetDeviceID(size_t device_id) { in SetDeviceID() function in mindspore::NNRTDeviceInfo
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_runtime.h104 void SetDeviceID(uint32_t device_id) { device_id_ = device_id; } in SetDeviceID() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/
Dcontext.cc164 void GPUDeviceInfo::SetDeviceID(uint32_t device_id) { in SetDeviceID() function in mindspore::GPUDeviceInfo
194 void AscendDeviceInfo::SetDeviceID(uint32_t device_id) { in SetDeviceID() function in mindspore::AscendDeviceInfo
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/
Dcontext.cc365 void GPUDeviceInfo::SetDeviceID(uint32_t device_id) { in SetDeviceID() function in mindspore::GPUDeviceInfo
391 void AscendDeviceInfo::SetDeviceID(uint32_t device_id) { in SetDeviceID() function in mindspore::AscendDeviceInfo