Searched refs:h2d_stream (Results 1 – 2 of 2) sorted by relevance
160 se::Stream* h2d_stream() { return gpu_->stream_->host_to_device; } in h2d_stream() function in tensorflow::GPUDeviceTestHelper383 gpu_helper_->h2d_stream()->ThenWaitFor(gpu_helper_->compute_stream()); in DoAddChain()388 gpu_helper_->h2d_stream()->ThenMemcpy( in DoAddChain()392 gpu_helper_->h2d_stream()->ThenMemcpy( in DoAddChain()394 gpu_helper_->compute_stream()->ThenWaitFor(gpu_helper_->h2d_stream()); in DoAddChain()
900 se::Stream* h2d_stream = local_device->host_to_device_stream(); in BufferFromHostLiteral() local903 h2d_stream, literal, buffer)); in BufferFromHostLiteral()908 local_device, std::move(device_buffer), event, h2d_stream)); in BufferFromHostLiteral()912 h2d_stream->RefreshStatus() in BufferFromHostLiteral()914 QCHECK(h2d_stream->ok()); in BufferFromHostLiteral()