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

Reduce duration of cpu_backend_gemm_test

Browse Source 
PiperOrigin-RevId: 247951547
This commit is contained in:
Jared Duke 2019-05-13 09:36:23 -07:00 committed by TensorFlower Gardener
parent fdd04bd83d
commit 2b4a08db39
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/kernels/cpu_backend_gemm_test.cc
View File

@ -594,7 +594,7 @@ TYPED_TEST(CpuBackendGemmTest, Rectangular) {
TYPED_TEST(CpuBackendGemmTest, HighlyRectangular) {
std::vector<std::tuple<int, int, int>> shapes;
for (int size = 1; size <= 10000; size *= 10) {
for (int size = 1; size <= 1000; size *= 10) {
shapes.push_back(std::make_tuple(size, 10, 10));
shapes.push_back(std::make_tuple(10, size, 10));
shapes.push_back(std::make_tuple(10, 10, size));