fix norm_op when batch != 1

src/operators/kernel/central-arm-func/norm_arm_func.h

@@ -56,9 +56,11 @@ void NormCompute(const NormParam<CPU> &param) {
   float *norm_ptr = norm->mutable_data<float>();
   float *out_ptr = out->mutable_data<float>();
 
+  for (int p = 0; p < pre; ++p) {
+    const float *in_tmp = input_ptr + p * n * post;
+    float *norm_tmp = norm_ptr + p * post;
+
     // in_ch = 0; norm = epsilon + x * x
-  const float *in_tmp = input_ptr;
-  float *norm_tmp = norm_ptr;
     for (int i = 0; i < post; ++i) {
       *norm_tmp = epsilon;
       *norm_tmp += (*in_tmp) * (*in_tmp);
@@ -67,8 +69,8 @@ void NormCompute(const NormParam<CPU> &param) {
     }
 
     // in_ch >= 1; norm += x * x
-  for (int j = 1; j < n; ++j) {
-    norm_tmp = norm_ptr;
+    for (int c = 1; c < n; ++c) {
+      norm_tmp = norm_ptr + p * post;
       for (int i = 0; i < post; ++i) {
         *norm_tmp += (*in_tmp) * (*in_tmp);
         norm_tmp++;
@@ -77,26 +79,23 @@ void NormCompute(const NormParam<CPU> &param) {
     }
 
     // norm = sqart(norm)
-  norm_tmp = norm_ptr;
+    norm_tmp = norm_ptr + p * post;
     for (int i = 0; i < post; ++i) {
-    float sqrt = sqrtf(*norm_tmp);
-    *norm_tmp = sqrt;
+      *norm_tmp = sqrtf(*norm_tmp);
       norm_tmp++;
     }
 
     // out = input / norm
-  in_tmp = input_ptr;
-  norm_tmp = norm_ptr;
-  float *out_tmp = out_ptr;
-  for (int i = 0; i < pre; ++i) {
-    for (int k = 0; k < n; ++k) {
+    in_tmp = input_ptr + p * n * post;
+    float *out_tmp = out_ptr + p * n * post;
+    for (int c = 0; c < n; ++c) {
+      norm_tmp = norm_ptr + p * post;
       for (int j = 0; j < post; ++j) {
         *out_tmp = *in_tmp / *norm_tmp;
         in_tmp++;
         norm_tmp++;
         out_tmp++;
       }
-      norm_tmp = norm_ptr + i * post;
     }
   }
 }