Searched refs:recv_host_to_device_stream (Results 1 – 1 of 1) sorted by relevance
314 auto recv_host_to_device_stream = in CopyCPUTensorToGPU() local317 if (recv_host_to_device_stream == nullptr) { in CopyCPUTensorToGPU()322 recv_host_to_device_stream->ThenWaitFor(recv_stream); in CopyCPUTensorToGPU()330 recv_host_to_device_stream->ThenMemcpy(&gpu_dst_ptr, src_ptr, total_bytes); in CopyCPUTensorToGPU()335 recv_host_to_device_stream, in CopyCPUTensorToGPU()336 [recv_host_to_device_stream, done, input_ref]() { in CopyCPUTensorToGPU()338 if (!recv_host_to_device_stream->ok()) { in CopyCPUTensorToGPU()