Searched defs:InitDevice (Results 1 – 7 of 7) sorted by relevance
29 bool GPUPsCache::InitDevice(uint32_t device_id, const void *) { in InitDevice() function in mindspore::ps::gpu::GPUPsCache
26 void GPUDeviceManager::InitDevice() { in InitDevice() function in mindspore::device::gpu::GPUDeviceManager
237 bool GPUKernelRuntime::InitDevice() { in InitDevice() function in mindspore::device::gpu::GPUKernelRuntime
30 bool InitDevice(int64_t device_num, int64_t global_rank, const std::string &backend, in InitDevice() function
110 bool AscendPsCache::InitDevice(uint32_t device_id, const void *context) { in InitDevice() function in mindspore::ps::ascend::AscendPsCache
111 bool GPUDeviceContext::InitDevice() { in InitDevice() function in mindspore::device::gpu::GPUDeviceContext
1003 bool AscendKernelRuntime::InitDevice() { in InitDevice() function in mindspore::device::ascend::AscendKernelRuntime