Searched refs:recv_host_to_device_stream (Results 1 – 1 of 1) sorted by relevance
313 auto recv_host_to_device_stream = in CopyCPUTensorToGPU() local316 if (recv_host_to_device_stream == nullptr) { in CopyCPUTensorToGPU()322 recv_host_to_device_stream->ThenWaitFor(recv_stream); in CopyCPUTensorToGPU()331 recv_host_to_device_stream->ThenMemcpy(&gpu_dst_ptr, src_ptr, total_bytes); in CopyCPUTensorToGPU()336 recv_host_to_device_stream, in CopyCPUTensorToGPU()337 [recv_host_to_device_stream, done, input_ref]() { in CopyCPUTensorToGPU()339 if (!recv_host_to_device_stream->ok()) { in CopyCPUTensorToGPU()