From b924be4e0344b03ebe1c16e50dc4fd717c8a7c6f Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Tue, 30 Jul 2019 12:14:58 -0700
Subject: [PATCH] Ruy - restore benchmark parameters

PiperOrigin-RevId: 260768613
---
 tensorflow/lite/experimental/ruy/benchmark.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow/lite/experimental/ruy/benchmark.cc b/tensorflow/lite/experimental/ruy/benchmark.cc
index 9aabacc37b9..7d055791a1e 100644
--- a/tensorflow/lite/experimental/ruy/benchmark.cc
+++ b/tensorflow/lite/experimental/ruy/benchmark.cc
@@ -73,9 +73,9 @@ void Benchmark() {
     setenv("QUICK_BENCHMARK", "1", 0);
 #endif
     std::vector<int> sizes;
-    for (int i = 16; i <= 2048; i *= 2) {
+    for (int i = 16; i <= 4096; i *= 2) {
       sizes.push_back(i);
-      if (i < 2048) {
+      if (i < 4096) {
         sizes.push_back(i * 3 / 2);
       }
     }