| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_basic.h | 35 virtual bool InitDevice(uint32_t device_id, const void *context) = 0;
|
| D | ps_cache_manager.cc | 330 embedding_device_cache_->cache_->InitDevice(device_id, context); in ProcessDataTask()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/ |
| D | gpu_ps_cache.h | 31 bool InitDevice(uint32_t device_id, const void *context) override;
|
| D | gpu_ps_cache.cc | 29 bool GPUPsCache::InitDevice(uint32_t device_id, const void *) { in InitDevice() function in mindspore::ps::gpu::GPUPsCache
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_device_manager.h | 33 void InitDevice();
|
| D | gpu_device_manager.cc | 26 void GPUDeviceManager::InitDevice() { in InitDevice() function in mindspore::device::gpu::GPUDeviceManager
|
| D | gpu_kernel_runtime.h | 67 bool InitDevice();
|
| D | gpu_kernel_runtime.cc | 87 bool ret = InitDevice(); in Init()
237 bool GPUKernelRuntime::InitDevice() { in InitDevice() function in mindspore::device::gpu::GPUKernelRuntime
248 GPUDeviceManager::GetInstance().InitDevice(); in InitDevice()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.h | 53 bool InitDevice(uint32_t device_id, const void *context) override;
|
| D | ascend_ps_cache.cc | 110 bool AscendPsCache::InitDevice(uint32_t device_id, const void *context) { in InitDevice() function in mindspore::ps::ascend::AscendPsCache
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.h | 76 bool InitDevice();
|
| D | gpu_device_context.cc | 83 if (!InitDevice()) { in Initialize()
111 bool GPUDeviceContext::InitDevice() { in InitDevice() function in mindspore::device::gpu::GPUDeviceContext
125 GPUDeviceManager::GetInstance().InitDevice(); in InitDevice()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | device_manager.h | 51 bool InitDevice(int64_t device_num, int64_t global_rank, const std::string &backend, const std::vec…
|
| D | device_manager.cc | 30 bool InitDevice(int64_t device_num, int64_t global_rank, const std::string &backend, in InitDevice() function
|
| D | step_parallel.cc | 2962 if (!InitDevice(device_num, global_rank, comm_info.communication_backend, stages)) { in ParallelInit()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.h | 85 bool InitDevice();
|
| D | ascend_kernel_runtime.cc | 344 bool ret = InitDevice(); in Init()
1003 bool AscendKernelRuntime::InitDevice() { in InitDevice() function in mindspore::device::ascend::AscendKernelRuntime
|