From bc66c88255a29460fb26dde0a8558db6a3524cd5 Mon Sep 17 00:00:00 2001 From: Eugene Zhulenev Date: Tue, 26 Nov 2019 11:41:57 -0800 Subject: Add async evaluation support to TensorPadding/TensorImagePatch/TensorShuffling --- unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h') diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h b/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h index 722032a3a..e6bb56136 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h @@ -176,7 +176,8 @@ struct TensorEvaluator, Device> /*dst_base=*/m_buffer + desc.offset(), /*dst_strides=*/internal::strides(m_impl.dimensions())); - ArgTensorBlock block = m_impl.blockV2(desc, scratch); + ArgTensorBlock block = + m_impl.blockV2(desc, scratch, /*root_of_expr_ast=*/true); // If block was evaluated into a destination buffer, there is no need to do // an assignment. -- cgit v1.2.3