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

Tweak test sizes for depthwise_conv

Browse Source 
PiperOrigin-RevId: 306915671
Change-Id: I5cfb74ab03aeb4caf2922df6c40470cb7ecec2c3
This commit is contained in:
Jared Duke 2020-04-16 13:57:06 -07:00 committed by TensorFlower Gardener
parent 203942391f
commit f10b9e1997
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tensorflow/lite/kernels/internal/depthwiseconv_quantized_test.cc
View File

@ -1089,7 +1089,7 @@ void TestOneNeonDot3x3(const TestParam& test_param) {
} }
TEST(TestDepthwiseConv, TestDepthwiseConv) { TEST(TestDepthwiseConv, TestDepthwiseConv) {
const int kTestsToRun = 10 * 1000; const int kTestsToRun = 1000;
for (int i = 0; i < kTestsToRun; i++) { for (int i = 0; i < kTestsToRun; i++) {
TestOneDepthwiseConv(DepthwiseConvImplementation::kNone, TestOneDepthwiseConv(DepthwiseConvImplementation::kNone,
DepthwiseConvOutputRounding::kAwayFromZero); DepthwiseConvOutputRounding::kAwayFromZero);
@ -1098,7 +1098,7 @@ TEST(TestDepthwiseConv, TestDepthwiseConv) {
// Run basic coverage test against the generic kernel. // Run basic coverage test against the generic kernel.
TEST(TestDepthwiseConv, TestGenericKernel) { TEST(TestDepthwiseConv, TestGenericKernel) {
const int kTestsToRun = 10 * 1000; const int kTestsToRun = 1000;
for (int i = 0; i < kTestsToRun; i++) { for (int i = 0; i < kTestsToRun; i++) {
TestOneDepthwiseConv(DepthwiseConvImplementation::kUseGenericKernel, TestOneDepthwiseConv(DepthwiseConvImplementation::kUseGenericKernel,
DepthwiseConvOutputRounding::kAwayFromZero); DepthwiseConvOutputRounding::kAwayFromZero);
@ -1107,7 +1107,7 @@ TEST(TestDepthwiseConv, TestGenericKernel) {
#if defined(__aarch64__) && !defined(GOOGLE_L4T) #if defined(__aarch64__) && !defined(GOOGLE_L4T)
TEST(TestDepthwiseConv, TestNeon3x3FilterAway) { TEST(TestDepthwiseConv, TestNeon3x3FilterAway) {
const int kTestsToRun = 1000; const int kTestsToRun = 500;
for (int i = 0; i < kTestsToRun; i++) { for (int i = 0; i < kTestsToRun; i++) {
TestOneDepthwiseConv3x3Filter(DepthwiseConvImplementation::kUseNeon3x3, TestOneDepthwiseConv3x3Filter(DepthwiseConvImplementation::kUseNeon3x3,
DepthwiseConvOutputRounding::kAwayFromZero); DepthwiseConvOutputRounding::kAwayFromZero);
@ -1115,7 +1115,7 @@ TEST(TestDepthwiseConv, TestNeon3x3FilterAway) {
} }
TEST(TestDepthwiseConv, TestNeon3x3FilterUpward) { TEST(TestDepthwiseConv, TestNeon3x3FilterUpward) {
const int kTestsToRun = 1000; const int kTestsToRun = 500;
for (int i = 0; i < kTestsToRun; i++) { for (int i = 0; i < kTestsToRun; i++) {
TestOneDepthwiseConv3x3Filter(DepthwiseConvImplementation::kUseNeon3x3, TestOneDepthwiseConv3x3Filter(DepthwiseConvImplementation::kUseNeon3x3,
DepthwiseConvOutputRounding::kUpward); DepthwiseConvOutputRounding::kUpward);
@ -1229,7 +1229,7 @@ INSTANTIATE_TEST_SUITE_P(
testing::Combine( testing::Combine(
Values(DepthwiseConvImplementation:: Values(DepthwiseConvImplementation::
kUseCModel3x3DotProduct), // forced_invocation kUseCModel3x3DotProduct), // forced_invocation
Values(1000), // tests_to_run Values(200), // tests_to_run
Values(QuantizationType::kNonPerChannelUint8), // quantization_type Values(QuantizationType::kNonPerChannelUint8), // quantization_type
Bool(), // test_stride Bool(), // test_stride
Bool(), // test_pad Bool(), // test_pad
@ -1245,7 +1245,7 @@ INSTANTIATE_TEST_SUITE_P(
testing::Combine( testing::Combine(
Values(DepthwiseConvImplementation:: Values(DepthwiseConvImplementation::
kUseUnwound3x3DotProduct), // forced_invocation kUseUnwound3x3DotProduct), // forced_invocation
Values(1000), // tests_to_run Values(200), // tests_to_run
Values(QuantizationType::kNonPerChannelUint8), // quantization_type Values(QuantizationType::kNonPerChannelUint8), // quantization_type
Bool(), // test_stride Bool(), // test_stride
Bool(), // test_pad Bool(), // test_pad
@ -1265,7 +1265,7 @@ INSTANTIATE_TEST_SUITE_P(
testing::Combine( testing::Combine(
Values(DepthwiseConvImplementation:: Values(DepthwiseConvImplementation::
kUseIntrinsics3x3DotProduct), // forced_invocation kUseIntrinsics3x3DotProduct), // forced_invocation
Values(500), // tests_to_run Values(200), // tests_to_run
Values(QuantizationType::kNonPerChannelUint8), // quantization_type Values(QuantizationType::kNonPerChannelUint8), // quantization_type
Bool(), // test_stride Bool(), // test_stride
Bool(), // test_pad Bool(), // test_pad
@ -1285,7 +1285,7 @@ INSTANTIATE_TEST_SUITE_P(
testing::Combine( testing::Combine(
Values(DepthwiseConvImplementation:: Values(DepthwiseConvImplementation::
kUseIntrinsics3x3DotProduct), // forced_invocation kUseIntrinsics3x3DotProduct), // forced_invocation
Values(500), // tests_to_run Values(200), // tests_to_run
Values(QuantizationType::kPerChannelInt8), // quantization_type Values(QuantizationType::kPerChannelInt8), // quantization_type
Bool(), // test_stride Bool(), // test_stride
Bool(), // test_pad Bool(), // test_pad
@ -1306,7 +1306,7 @@ INSTANTIATE_TEST_SUITE_P(
testing::Combine( testing::Combine(
Values(DepthwiseConvImplementation:: Values(DepthwiseConvImplementation::
kUseNeon3x3DotProduct), // forced_invocation kUseNeon3x3DotProduct), // forced_invocation
Values(1000), // tests_to_run Values(200), // tests_to_run
Values(QuantizationType::kNonPerChannelUint8), // quantization_type Values(QuantizationType::kNonPerChannelUint8), // quantization_type
Bool(), // test_stride Bool(), // test_stride
Bool(), // test_pad Bool(), // test_pad
@ -1322,7 +1322,7 @@ INSTANTIATE_TEST_SUITE_P(
testing::Combine( testing::Combine(
Values(DepthwiseConvImplementation:: Values(DepthwiseConvImplementation::
kUseNeon3x3DotProduct), // forced_invocation kUseNeon3x3DotProduct), // forced_invocation
Values(1000), // tests_to_run Values(200), // tests_to_run
Values(QuantizationType::kPerChannelInt8), // quantization_type Values(QuantizationType::kPerChannelInt8), // quantization_type
Bool(), // test_stride Bool(), // test_stride
Bool(), // test_pad Bool(), // test_pad
@ -1339,7 +1339,7 @@ INSTANTIATE_TEST_SUITE_P(
Dispatch3x3, DepthwiseConvTest, Dispatch3x3, DepthwiseConvTest,
testing::Combine( testing::Combine(
Values(DepthwiseConvImplementation::kNone), // forced_invocation Values(DepthwiseConvImplementation::kNone), // forced_invocation
Values(500), // tests_to_run Values(200), // tests_to_run
Values(QuantizationType::kNonPerChannelUint8), // quantization_type Values(QuantizationType::kNonPerChannelUint8), // quantization_type
Bool(), // test_stride Bool(), // test_stride
Bool(), // test_pad Bool(), // test_pad
@ -1354,7 +1354,7 @@ INSTANTIATE_TEST_SUITE_P(
Dispatch3x3PerChannel, DepthwiseConvTest, Dispatch3x3PerChannel, DepthwiseConvTest,
testing::Combine( testing::Combine(
Values(DepthwiseConvImplementation::kNone), // forced_invocation Values(DepthwiseConvImplementation::kNone), // forced_invocation
Values(500), // tests_to_run Values(200), // tests_to_run
Values(QuantizationType::kPerChannelInt8), // quantization_type Values(QuantizationType::kPerChannelInt8), // quantization_type
Bool(), // test_stride Bool(), // test_stride
Bool(), // test_pad Bool(), // test_pad