Searched refs:d2h_stream (Results 1 – 1 of 1) sorted by relevance
161 se::Stream* d2h_stream() { return gpu_->stream_->device_to_host; } in d2h_stream() function in tensorflow::GPUDeviceTestHelper418 gpu_helper_->d2h_stream()->ThenWaitFor(gpu_helper_->compute_stream()); in DoAddChain()422 gpu_helper_->d2h_stream()->ThenMemcpy(DMAHelper::base(&host_outputs_[0]), in DoAddChain()424 gpu_helper_->event_mgr()->ThenExecute(gpu_helper_->d2h_stream(), in DoAddChain()428 gpu_helper_->d2h_stream(), [times, r]() { in DoAddChain()