commit | 9be0c5a71cd7c1f08b65c48e53d083c59dd093a3 | [log] [tgz] |
---|---|---|
author | Georgios Pinitas <georgios.pinitas@arm.com> | Mon Feb 19 12:46:29 2018 +0000 |
committer | Anthony Barbier <anthony.barbier@arm.com> | Fri Nov 02 16:47:18 2018 +0000 |
tree | f9029878deaa3714dbded630bf3c48eb9af06623 | |
parent | 287b570b86ba40a801136aded140b83435ca9314 [diff] [blame] |
COMPMID-936: Convolution failure in NEON Convolution Layer. Change-Id: I68a98eff57c8db719a501b68541666e8bc5f2081 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/121180 Tested-by: Jenkins <bsgcomp@arm.com> Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp b/src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp index 3f33c43..08d8f8c 100644 --- a/src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp +++ b/src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp
@@ -451,8 +451,10 @@ break; } case DataType::S32: + { _func = (bias == nullptr) ? &output_stage<int32_t, uint8_t, false, false> : &output_stage<int32_t, uint8_t, false, true>; break; + } #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC case DataType::F16: {