Searched defs:CopyDeviceMemToHost (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/ |
| D | gpu_ps_cache.cc | 76 bool GPUPsCache::CopyDeviceMemToHost(void *dst, const void *src, size_t size) { in CopyDeviceMemToHost() function in mindspore::ps::gpu::GPUPsCache
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_device_manager.cc | 106 bool GPUDeviceManager::CopyDeviceMemToHost(const HostMemPtr &dst, const DeviceMemPtr &src, size_t s… in CopyDeviceMemToHost() function in mindspore::device::gpu::GPUDeviceManager
|
| D | cuda_driver.cc | 85 bool CudaDriver::CopyDeviceMemToHost(const HostMemPtr &dst, const DeviceMemPtr &src, size_t size) { in CopyDeviceMemToHost() function in mindspore::device::gpu::CudaDriver
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.cc | 201 bool AscendPsCache::CopyDeviceMemToHost(void *dst, const void *src, size_t size) { in CopyDeviceMemToHost() function in mindspore::ps::ascend::AscendPsCache
|
| /third_party/mindspore/mindspore/lite/src/runtime/gpu/opencl/ |
| D | opencl_runtime.cc | 556 bool OpenCLRuntime::CopyDeviceMemToHost(void *dst, const void *src, size_t size, cl::CommandQueue *… in CopyDeviceMemToHost() function in mindspore::lite::opencl::OpenCLRuntime
|