diff options
author | Tim Shen <timshen@google.com> | 2018-08-30 11:20:16 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-30 11:29:03 -0700 |
commit | 09f2c342c0e74834bebf8045d5e77dcef8323539 (patch) | |
tree | 3a969695f54840366629df7f39e4da7f9ac103cd /tensorflow/stream_executor | |
parent | 9e12f1df3270b5e0b310645e6c3cae9fbd3f5dfc (diff) |
Remove (Mutable)ArraySlice implementation and alias them to absl::Span.
There are several API migrations happening:
* ArraySlice's sub-slice constructor => .subspan
* MutableArraySlice's container pointer constructor => absl::MakeSpan
PiperOrigin-RevId: 210946124
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; |