From 34a75c3c5cec4e2bfe5c68164f8c3372f6ae5ecb Mon Sep 17 00:00:00 2001 From: Eugene Zhulenev Date: Fri, 20 Jul 2018 17:37:20 -0700 Subject: Initial support of TensorBlock --- unsupported/Eigen/CXX11/src/Tensor/TensorDimensions.h | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorDimensions.h') diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorDimensions.h b/unsupported/Eigen/CXX11/src/Tensor/TensorDimensions.h index 86405e69b..192d4aa7b 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorDimensions.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorDimensions.h @@ -284,6 +284,12 @@ struct DSizes : array { (*this)[0] = i0; } + EIGEN_DEVICE_FUNC DSizes(const DimensionList& a) { + for (int i = 0 ; i < NumDims; ++i) { + (*this)[i] = a[i]; + } + } + #if EIGEN_HAS_VARIADIC_TEMPLATES template EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE explicit DSizes(DenseIndex firstDimension, DenseIndex secondDimension, IndexTypes... otherDimensions) : Base({{firstDimension, secondDimension, otherDimensions...}}) { -- cgit v1.2.3