Searched refs:send_device_to_device_stream (Results 1 – 1 of 1) sorted by relevance
200 auto send_device_to_device_stream = in DeviceToDeviceCopy() local203 if (send_device_to_device_stream == nullptr) { in DeviceToDeviceCopy()209 send_device_to_device_stream->ThenWaitFor(send_stream); in DeviceToDeviceCopy()228 send_device_to_device_stream->ThenWaitFor(recv_stream); in DeviceToDeviceCopy()231 send_device_to_device_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, in DeviceToDeviceCopy()238 send_device_to_device_stream, in DeviceToDeviceCopy()239 [done, send_device_to_device_stream, input_ref]() { in DeviceToDeviceCopy()241 if (!send_device_to_device_stream->ok()) { in DeviceToDeviceCopy()247 send_device_to_device_stream); in DeviceToDeviceCopy()