experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #28744 from samikama:GPUTestOutputFix

Browse Source 
PiperOrigin-RevId: 248748011
This commit is contained in:
TensorFlower Gardener 2019-05-17 10:51:26 -07:00
commit 27f15b4998
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/ops_testutil.cc
View File

@ -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;
}