From ffb78e23a1b3bc232a07773144cfa5fa1759852d Mon Sep 17 00:00:00 2001
From: Antonio Sanchez <cantonios@google.com>
Date: Mon, 15 Nov 2021 22:19:04 -0800
Subject: [PATCH] Fix tensor broadcast off-by-one error.

Caught by JAX unit tests.  Triggered if broadcast is smaller than packet
size.
---
 unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h
index 58d3cc29b..ab6dcb678 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h
@@ -463,8 +463,8 @@ struct TensorEvaluator<const TensorBroadcastingOp<Broadcast, ArgType>, Device>
           values[i] = m_impl.coeff(inputIndex);
           ++outputOffset;
         } else {
-          outputOffset = 0;
           values[i] = m_impl.coeff(++inputIndex);
+          outputOffset = 1;  // Next offset.
         }
       }
       return internal::pload<PacketReturnType>(values);