Searched refs:gpu_dst_ptr (Results 1 – 1 of 1) sorted by relevance
217 DeviceMemoryBase gpu_dst_ptr(dst_ptr, total_bytes); in DeviceToDeviceCopy() local232 send_device_to_device_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, in DeviceToDeviceCopy()329 DeviceMemoryBase gpu_dst_ptr(dst_ptr, total_bytes); in CopyCPUTensorToGPU() local330 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()