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

Reduce the number of tests inside unit-test and shard it to speed it up.

Browse Source 
PiperOrigin-RevId: 228373086
This commit is contained in:
Karim Nosir 2019-01-08 11:59:53 -08:00 committed by TensorFlower Gardener
parent 35279e5c61
commit a849894f02
3 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/kernels/internal/BUILD
View File

@ -641,6 +641,7 @@ cc_test(
srcs = [
"softmax_quantized_test.cc",
],
shard_count = 3,
deps = [
":optimized_base",
":quantization_util",
@ -657,6 +658,7 @@ cc_test(
srcs = [
"logsoftmax_quantized_test.cc",
],
shard_count = 3,
tags = [
# TODO(b/122242739): Reenable after fixing the flakiness?
"nomac",

6
tensorflow/lite/kernels/internal/logsoftmax_quantized_test.cc
View File

@ -225,7 +225,7 @@ bool TryOneSkyscraperLogSoftmax(bool small_depth) {
}
TEST(TestQuantizedLogSoftmax, UniformLogSoftmaxTests) {
const int kTestsToRun = 1000;
const int kTestsToRun = 100;
for (int i = 0; i < kTestsToRun; i++) {
while (!TryOneUniformLogSoftmax()) {
}
@ -233,7 +233,7 @@ TEST(TestQuantizedLogSoftmax, UniformLogSoftmaxTests) {
}
TEST(TestQuantizedLogSoftmax, SkyscraperLogSoftmaxTests) {
const int kTestsToRun = 1000;
const int kTestsToRun = 100;
for (int i = 0; i < kTestsToRun; i++) {
while (!TryOneSkyscraperLogSoftmax(false)) {
}
@ -241,7 +241,7 @@ TEST(TestQuantizedLogSoftmax, SkyscraperLogSoftmaxTests) {
}
TEST(TestQuantizedLogSoftmax, SmallSkyscraperLogSoftmaxTests) {
const int kTestsToRun = 1000;
const int kTestsToRun = 100;
for (int i = 0; i < kTestsToRun; i++) {
while (!TryOneSkyscraperLogSoftmax(true)) {
}

6
tensorflow/lite/kernels/internal/softmax_quantized_test.cc
View File

@ -210,7 +210,7 @@ bool TryOneSkyscraperSoftmax(bool small_depth) {
}
TEST(TestQuantizedSoftmax, UniformSoftmaxTests) {
const int kTestsToRun = 1000;
const int kTestsToRun = 100;
for (int i = 0; i < kTestsToRun; i++) {
while (!TryOneUniformSoftmax()) {
}
@ -218,7 +218,7 @@ TEST(TestQuantizedSoftmax, UniformSoftmaxTests) {
}
TEST(TestQuantizedSoftmax, SkyscraperSoftmaxTests) {
const int kTestsToRun = 1000;
const int kTestsToRun = 100;
for (int i = 0; i < kTestsToRun; i++) {
while (!TryOneSkyscraperSoftmax(false)) {
}
@ -226,7 +226,7 @@ TEST(TestQuantizedSoftmax, SkyscraperSoftmaxTests) {
}
TEST(TestQuantizedSoftmax, SmallSkyscraperSoftmaxTests) {
const int kTestsToRun = 1000;
const int kTestsToRun = 100;
for (int i = 0; i < kTestsToRun; i++) {
while (!TryOneSkyscraperSoftmax(true)) {
}