Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_utils.cc545 int SetCudaDevice(std::shared_ptr<GPUDeviceInfo> device_info_) { in SetCudaDevice() function
549 int SetCudaDevice(int device_id) { in SetCudaDevice() function