提交 d30b11a0 编写于 作者: L liuqi 提交者: wuchenghui

Update the batchnorm test and benchmark.

上级 dd2b603d
......@@ -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);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册