diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2019-11-26 11:41:57 -0800 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-11-26 11:41:57 -0800 |
| commit | bc66c88255a29460fb26dde0a8558db6a3524cd5 (patch) | |
| tree | fb0d883d7abc904db7080b842b34488a6c02a23c /unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h | |
| parent | c79b6ffe1fcf8c12005942a1268f79b7d6ecf700 (diff) | |
Add async evaluation support to TensorPadding/TensorImagePatch/TensorShuffling
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h | 3 |
1 files changed, 2 insertions, 1 deletions
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<const TensorEvalToOp<ArgType, MakePointer_>, Device> /*dst_base=*/m_buffer + desc.offset(), /*dst_strides=*/internal::strides<Layout>(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. |