experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Ruy - restore benchmark parameters

Browse Source 
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
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/experimental/ruy/benchmark.cc
View File

@ -73,9 +73,9 @@ void Benchmark() {
setenv("QUICK_BENCHMARK", "1", 0); setenv("QUICK_BENCHMARK", "1", 0);
#endif #endif
std::vector<int> sizes; std::vector<int> sizes;
for (int i = 16; i <= 2048; i *= 2) { for (int i = 16; i <= 4096; i *= 2) {
sizes.push_back(i); sizes.push_back(i);
if (i < 2048) { if (i < 4096) {
sizes.push_back(i * 3 / 2); sizes.push_back(i * 3 / 2);
} }
} }