Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/
Dgpu_ps_cache.cc76 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/
Dgpu_device_manager.cc106 bool GPUDeviceManager::CopyDeviceMemToHost(const HostMemPtr &dst, const DeviceMemPtr &src, size_t s… in CopyDeviceMemToHost() function in mindspore::device::gpu::GPUDeviceManager
Dcuda_driver.cc85 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/
Dascend_ps_cache.cc201 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/
Dopencl_runtime.cc556 bool OpenCLRuntime::CopyDeviceMemToHost(void *dst, const void *src, size_t size, cl::CommandQueue *… in CopyDeviceMemToHost() function in mindspore::lite::opencl::OpenCLRuntime