diff --git a/tensorflow/core/kernels/ops_testutil.cc b/tensorflow/core/kernels/ops_testutil.cc
index 7aa7d1a5861..e08abd2e03e 100644
--- a/tensorflow/core/kernels/ops_testutil.cc
+++ b/tensorflow/core/kernels/ops_testutil.cc
@@ -53,8 +53,8 @@ Tensor* OpsTestBase::GetOutput(int output_index) {
           new Tensor(allocator(), output->dtype(), output->shape());
       auto src = output->tensor_data();
       auto dst = managed_output->tensor_data();
-      context_->eigen_gpu_device().memcpy(const_cast<char*>(dst.data()),
-                                          src.data(), src.size());
+      context_->eigen_gpu_device().memcpyDeviceToHost(
+          const_cast<char*>(dst.data()), src.data(), src.size());
       context_->eigen_gpu_device().synchronize();
       managed_outputs_[output_index] = managed_output;
     }