Ruy - restore benchmark parameters

PiperOrigin-RevId: 260768613
This commit is contained in:
A. Unique TensorFlower 2019-07-30 12:14:58 -07:00 committed by TensorFlower Gardener
parent 0b9d4fd92d
commit b924be4e03

View File

@ -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);
}
}