From dd2b603d367f704e0e2c52a940eb037e32bad158 Mon Sep 17 00:00:00 2001 From: liuqi Date: Fri, 8 Sep 2017 15:14:24 +0800 Subject: [PATCH] Use float to instantiate the batch_norm neon. --- mace/kernels/neon/batch_norm_neon.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mace/kernels/neon/batch_norm_neon.cc b/mace/kernels/neon/batch_norm_neon.cc index 61cbed1a..244d564b 100644 --- a/mace/kernels/neon/batch_norm_neon.cc +++ b/mace/kernels/neon/batch_norm_neon.cc @@ -26,8 +26,8 @@ void BatchNormFunctor::operator()(const float* input, // new_offset = \offset - mean * common_val; // Y = new_scale * X + new_offset; float new_scale, new_offset; - int count = sample_size >> 2; - int remain_count = sample_size - count; + index_t count = sample_size >> 2; + index_t remain_count = sample_size - (count << 2); for (index_t c = 0; c < channel; ++c) { new_scale = scale[c] / std::sqrt(var[c] + this->variance_epsilon_); new_offset = offset[c] - mean[c] * new_scale; -- GitLab