diff --git a/tensorflow/core/kernels/lrn_op_test.cc b/tensorflow/core/kernels/lrn_op_test.cc
index 7604a40d029..68aa3428399 100644
--- a/tensorflow/core/kernels/lrn_op_test.cc
+++ b/tensorflow/core/kernels/lrn_op_test.cc
@@ -67,7 +67,7 @@ class LRNFloatTest : public OpsTestBase {
 
     Eigen::Tensor<float, 4, Eigen::RowMajor> expected(batch_size, rows, cols,
                                                       depth);
-    auto out = expected.reshape(Eigen::DSizes<int64, 2>{rest, depth});
+    auto out = expected.reshape(Eigen::DSizes<Eigen::Index, 2>{rest, depth});
     auto in = input.shaped<float, 2>({rest, depth});
 
     for (int64 i = 0; i < rest; ++i) {