Use DenseIndex in the MeanReducer to avoid overflows when processing very large tensors.

2025-01-24 14:45:14 +08:00 · 2016-04-19 11:53:58 -07:00 · 2016-04-19 11:53:58 -07:00 · 50968a0a3e
commit 50968a0a3e
parent 84543c8be2
1 changed files with 2 additions and 2 deletions
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h
@ -158,8 +158,8 @@ template <typename T> struct MeanReducer
  }

  protected:
-    int scalarCount_;
-    int packetCount_;
+    DenseIndex scalarCount_;
+    DenseIndex packetCount_;
 };

 template <typename T> struct MaxReducer