Revert the cpu and noen benchmark.

918aae3b · liuqi · ed99a2d1 · 918aae3b
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

mace/ops/batch_norm_benchmark.cc mace/ops/batch_norm_benchmark.cc +2 -0

未找到文件。
--- a/mace/ops/batch_norm_benchmark.cc
+++ b/mace/ops/batch_norm_benchmark.cc
@@ -60,6 +60,8 @@ static void BatchNorm(
  BENCHMARK(BM_BATCH_NORM_##N##_##C##_##H##_##W##_##TYPE##_##DEVICE)

 #define BM_BATCH_NORM(N, C, H, W, TYPE)       \
+  BM_BATCH_NORM_MACRO(N, C, H, W, TYPE, CPU); \
+  BM_BATCH_NORM_MACRO(N, C, H, W, TYPE, NEON);\
  BM_BATCH_NORM_MACRO(N, C, H, W, TYPE, OPENCL);

 BM_BATCH_NORM(1, 1, 512, 512, float);