Searched refs:send_device_to_device_stream (Results 1 – 1 of 1) sorted by relevance
201 auto send_device_to_device_stream = in DeviceToDeviceCopy() local204 if (send_device_to_device_stream == nullptr) { in DeviceToDeviceCopy()210 send_device_to_device_stream->ThenWaitFor(send_stream); in DeviceToDeviceCopy()229 send_device_to_device_stream->ThenWaitFor(recv_stream); in DeviceToDeviceCopy()232 send_device_to_device_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, in DeviceToDeviceCopy()239 send_device_to_device_stream, in DeviceToDeviceCopy()240 [done, send_device_to_device_stream, input_ref]() { in DeviceToDeviceCopy()242 if (!send_device_to_device_stream->ok()) { in DeviceToDeviceCopy()248 send_device_to_device_stream); in DeviceToDeviceCopy()