From d16634c4d47bf751ff55e884cd76f742a7fb8e17 Mon Sep 17 00:00:00 2001
From: Rasmus Munk Larsen <rmlarsen@google.com>
Date: Mon, 8 Oct 2018 16:41:36 -0700
Subject: Fix out-of bounds access in TensorArgMax.h.

---
 unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h')

diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h b/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h
index ea3ab329d..6f7c6d86d 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h
@@ -235,7 +235,6 @@ struct TensorEvaluator<const TensorTupleReducerOp<ReduceOp, Dims, ArgType>, Devi
        ,m_device(device)
 #endif
   {
-
     gen_strides(m_orig_impl.dimensions(), m_strides);
     if (Layout == static_cast<int>(ColMajor)) {
       const Index total_size = internal::array_prod(m_orig_impl.dimensions());
@@ -244,7 +243,7 @@ struct TensorEvaluator<const TensorTupleReducerOp<ReduceOp, Dims, ArgType>, Devi
       const Index total_size = internal::array_prod(m_orig_impl.dimensions());
       m_stride_mod = (m_return_dim > 0) ? m_strides[m_return_dim - 1] : total_size;
     }
-    m_stride_div = m_strides[m_return_dim];
+    m_stride_div = (m_return_dim >= 0) ? m_strides[m_return_dim] : 1;
   }
 
   EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const {
-- 
cgit v1.2.3