diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2019-10-16 17:14:37 -0700 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-10-16 17:14:37 -0700 |
| commit | 0d2a14ce11c85abdfc68ca37fc66e3cace088b24 (patch) | |
| tree | b7838dd2a8dfe6a36f2c85bf9af58c2048df5e5b /unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h | |
| parent | 02431cbe71eb036b1d6caa49c585db92a20b030f (diff) | |
Cleanup Tensor block destination and materialized block storage allocation
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h b/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h index 8c44f1c4a..22fc64c1f 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h @@ -173,12 +173,9 @@ struct TensorEvaluator<const TensorEvalToOp<ArgType, MakePointer_>, Device> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void evalBlockV2( TensorBlockDesc& desc, TensorBlockScratch& scratch) { // Add `m_buffer` as destination buffer to the block descriptor. - desc.AddDestinationBuffer( + desc.template AddDestinationBuffer<Layout>( /*dst_base=*/m_buffer + desc.offset(), - /*dst_strides=*/internal::strides<Layout>(m_impl.dimensions()), - /*total_dst_bytes=*/ - (internal::array_prod(m_impl.dimensions()) - * sizeof(Scalar))); + /*dst_strides=*/internal::strides<Layout>(m_impl.dimensions())); ArgTensorBlock block = m_impl.blockV2(desc, scratch); |