diff options
author | Eugene Zhulenev <ezhulenev@google.com> | 2019-10-07 15:34:26 -0700 |
---|---|---|
committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-10-07 15:34:26 -0700 |
commit | f74ab8cb8de5e425ddd25f4b06657926a2ad4599 (patch) | |
tree | 21686c69f54cd402fdf6508cedcfd25750f70898 /unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | |
parent | 3afb640b5647654f272b1903b71877cb60ed3a78 (diff) |
Add block evaluation to TensorEvalTo and fix few small bugs
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h index a7cb8dc97..97ac96db1 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h @@ -238,7 +238,8 @@ class TensorExecutor<Expression, DefaultDevice, Vectorizable, typedef TensorBlockMapper<ScalarNoConst, StorageIndex, NumDims, Evaluator::Layout> TensorBlockMapper; typedef typename TensorBlock::Dimensions TensorBlockDimensions; - typedef internal::TensorBlockDescriptor<NumDims> TensorBlockDesc; + typedef internal::TensorBlockDescriptor<NumDims, StorageIndex> + TensorBlockDesc; typedef internal::TensorBlockScratchAllocator<DefaultDevice> TensorBlockScratch; |