Browse Source

fix build, second try

tags/20200916
nihui 6 years ago
parent
commit
9fe137a3c2
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      src/layer/arm/batchnorm_arm.cpp

+ 3
- 1
src/layer/arm/batchnorm_arm.cpp View File

@@ -153,10 +153,11 @@ int BatchNorm_arm::forward_inplace(Mat& bottom_top_blob, const Option& opt) cons
float a = a_data[i];
float b = b_data[i];

int j = 0;
#if __ARM_NEON
float32x4_t _a = vdupq_n_f32(a);
float32x4_t _b = vdupq_n_f32(b);

int j = 0;
for (; j + 3 < w; j += 4)
{
float32x4_t _p = vld1q_f32(ptr);
@@ -165,6 +166,7 @@ int BatchNorm_arm::forward_inplace(Mat& bottom_top_blob, const Option& opt) cons

ptr += 4;
}
#endif // __ARM_NEON
for (; j < w; j++)
{
*ptr = b * *ptr + a;


Loading…
Cancel
Save