Searched refs:gpu_dst_ptr (Results 1 – 1 of 1) sorted by relevance
216 DeviceMemoryBase gpu_dst_ptr(dst_ptr, total_bytes); in DeviceToDeviceCopy() local231 send_device_to_device_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, in DeviceToDeviceCopy()330 DeviceMemoryBase gpu_dst_ptr(dst_ptr, total_bytes); in CopyCPUTensorToGPU() local331 recv_host_to_device_stream->ThenMemcpy(&gpu_dst_ptr, src_ptr, total_bytes); in CopyCPUTensorToGPU()442 DeviceMemoryBase gpu_dst_ptr(dst_ptr, total_bytes); in CopyGPUTensorToSameGPU() local443 send_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, total_bytes); in CopyGPUTensorToSameGPU()