From 2d818f3658bfd3302e0afcd9d6c65973f814360a Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Tue, 13 Aug 2019 09:31:29 +0000 Subject: [PATCH] Fix sign mismatch in NEON mean Signed-off-by: Matthew Bentham --- tensorflow/lite/kernels/internal/optimized/integer_ops/mean.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/lite/kernels/internal/optimized/integer_ops/mean.h b/tensorflow/lite/kernels/internal/optimized/integer_ops/mean.h index 4afec531ebe..0c1ab818199 100644 --- a/tensorflow/lite/kernels/internal/optimized/integer_ops/mean.h +++ b/tensorflow/lite/kernels/internal/optimized/integer_ops/mean.h @@ -103,7 +103,7 @@ inline void MeanImpl(const tflite::MeanParams& op_params, int32x4_t casted_mean_2 = RoundToNearest(mean_2); int16x4_t narrow_range_mean_2 = vmovn_s32(casted_mean_2); int16x8_t combined_mean = - vcombine_u16(narrow_range_mean_2, narrow_range_mean_1); + vcombine_s16(narrow_range_mean_2, narrow_range_mean_1); int8x8_t narrowed_combined_mean = vmovn_s16(combined_mean); int8_t* output_data_ptr = output_data + Offset(output_shape, out_b, 0, 0, out_d);