Lines Matching defs:send_stream
123 gpu::Stream* send_stream = nullptr;
125 &send_stream);
139 send_device_to_host_stream->ThenWaitFor(send_stream);
198 gpu::Stream* send_stream = nullptr;
200 &send_stream);
214 send_device_to_device_stream->ThenWaitFor(send_stream);
265 gpu::Stream* send_stream = nullptr;
267 &dev_info, &send_stream);
281 send_device_to_host_stream->ThenWaitFor(send_stream);
433 gpu::Stream* send_stream = nullptr;
435 dst_gpu_tensor, &dev_info, &send_stream);
447 send_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, total_bytes);
