diff options
Diffstat (limited to 'tensorflow/stream_executor')
-rw-r--r-- | tensorflow/stream_executor/cuda/cuda_dnn.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/cuda/cuda_dnn.cc b/tensorflow/stream_executor/cuda/cuda_dnn.cc index 55408ab9ab..207f22c931 100644 --- a/tensorflow/stream_executor/cuda/cuda_dnn.cc +++ b/tensorflow/stream_executor/cuda/cuda_dnn.cc @@ -3894,7 +3894,7 @@ bool CudnnSupport::DoDepthConcatenate( for (size_t i = 0; i < input_data.size(); ++i) { const auto& dimensions = input_dimensions[i]; tmp.resize(dimensions.ElementCount()); - stream->ThenMemcpyD2H<float>(*input_data[i], &tmp); + stream->ThenMemcpyD2H<float>(*input_data[i], absl::MakeSpan(tmp)); port::Status block_status = stream->BlockHostUntilDone(); if (!block_status.ok()) { LOG(ERROR) << "BlockHostUntilDone failed: " << block_status; |