From d30b11a02c2e4c545fbf6937c395d6e1bc7b9cf8 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Fri, 8 Sep 2017 16:34:00 +0800
Subject: [PATCH] Update the batchnorm test and benchmark.

---
 mace/kernels/test/batch_norm_neon_test.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mace/kernels/test/batch_norm_neon_test.cc b/mace/kernels/test/batch_norm_neon_test.cc
index b33b316c..b47a3360 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<float> input(input_size, 0.0);
-- 
GitLab