From ac88b10f3fbe258e9322cd6913b4678574b4e57f Mon Sep 17 00:00:00 2001 From: Jian Li Date: Wed, 27 Mar 2019 10:55:01 -0700 Subject: [PATCH] Increase maxpool test depth. Depth larger than kPoolingAccTrancheSize is needed to properly test the neon code. PiperOrigin-RevId: 240597128 --- tensorflow/lite/kernels/internal/maxpool_quantized_test.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/lite/kernels/internal/maxpool_quantized_test.cc b/tensorflow/lite/kernels/internal/maxpool_quantized_test.cc index 6b6a23e1229..c16c8608e7e 100644 --- a/tensorflow/lite/kernels/internal/maxpool_quantized_test.cc +++ b/tensorflow/lite/kernels/internal/maxpool_quantized_test.cc @@ -57,7 +57,7 @@ void RunOneMaxPoolTest(const PoolParams& params, // test function. void CreateDataAndRunMaxPool(bool padding_same) { const int batch = UniformRandomInt(1, 20); - const int input_depth = UniformRandomInt(32, 128); + const int input_depth = UniformRandomInt(32, 700); const int output_depth = input_depth; const int input_width = UniformRandomInt(64, 128); const int input_height = UniformRandomInt(64, 128);