From 7e3bc4177e6ab4c6d4e16dfabca5c6f9290d8441 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.


(cherry picked from commit ffb78e23a1b3bc232a07773144cfa5fa1759852d)
---
 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 8d8ad2658..7449b046b 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h
@@ -461,8 +461,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);