diff --git a/mace/kernels/test/batch_norm_neon_test.cc b/mace/kernels/test/batch_norm_neon_test.cc index b33b316c109fe303d0e99e3e663f88712e0d770c..b47a3360bc33d850eb63929b4dc9c800ba8419b6 100644 --- a/mace/kernels/test/batch_norm_neon_test.cc +++ b/mace/kernels/test/batch_norm_neon_test.cc @@ -17,8 +17,8 @@ TEST(BatchNormNeonTest, Simple) { // generate random input index_t batch = 1 + rand() % 128; index_t channels = 3; - index_t height = 2 + rand() % 100; - index_t width = 2 + rand() % 100; + index_t height = 10 + rand() % 100; + index_t width = 10 + rand() % 100; index_t input_size = batch * channels * height * width; std::vector input(input_size, 0.0);