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/TensorAssign.h | |
parent | 02431cbe71eb036b1d6caa49c585db92a20b030f (diff) |
Cleanup Tensor block destination and materialized block storage allocation
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorAssign.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorAssign.h | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorAssign.h b/unsupported/Eigen/CXX11/src/Tensor/TensorAssign.h index f2b9389c8..1f64de3a9 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorAssign.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorAssign.h @@ -235,11 +235,9 @@ struct TensorEvaluator<const TensorAssignOp<LeftArgType, RightArgType>, Device> m_leftImpl.data() != NULL) { // If destination has raw data access, we pass it as a potential // destination for a block descriptor evaluation. - desc.AddDestinationBuffer( + desc.template AddDestinationBuffer<Layout>( /*dst_base=*/m_leftImpl.data() + desc.offset(), - /*dst_strides=*/internal::strides<Layout>(m_leftImpl.dimensions()), - /*total_dst_bytes=*/ - (internal::array_prod(m_leftImpl.dimensions()) * sizeof(Scalar))); + /*dst_strides=*/internal::strides<Layout>(m_leftImpl.dimensions())); } RightTensorBlock block = m_rightImpl.blockV2(desc, scratch, /*root_of_expr_ast=*/true); |