experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix sign mismatch in NEON mean

Browse Source 
Signed-off-by: Matthew Bentham <Matthew.Bentham@arm.com>
This commit is contained in:
Matthew Bentham 2019-08-13 09:31:29 +00:00
parent f20b214cd0
commit 2d818f3658
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/kernels/internal/optimized/integer_ops/mean.h
View File

@ -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);