diff options
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h index 3b2100ab0..d96303224 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorFunctors.h @@ -372,7 +372,7 @@ template <typename T> struct ArgMaxTupleReducer EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const T t, T* accum) const { if (t.second < accum->second) { return; - } else if (t.second > accum->second || t.first < accum->first) { + } else if (t.second > accum->second || accum->first > t.first ) { *accum = t; } } @@ -400,7 +400,7 @@ template <typename T> struct ArgMinTupleReducer EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void reduce(const T& t, T* accum) const { if (t.second > accum->second) { return; - } else if (t.second < accum->second || t.first < accum->first) { + } else if (t.second < accum->second || accum->first > t.first) { *accum = t; } } |