diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-29 13:58:31 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-29 16:36:56 +0000 |
commit | 551852fb43a5cf7ed9ac8679517b103e9af8a38f (patch) | |
tree | 6d2d3fc7349c9153101684722e520d5fa99d5eb6 /src/core | |
parent | 618493d9823936799501334d06572c5f2d8da319 (diff) | |
download | ComputeLibrary-551852fb43a5cf7ed9ac8679517b103e9af8a38f.tar.gz |
COMPMID-1451: Remove invalid bias inclusion in specialized function.
Fixed Segfault by removing bias iterator from specialized function which assumes no bias
being provided.
Change-Id: Ic897435ee9427d4359e8ab989a03e951da0d7ce0
Reviewed-on: https://review.mlplatform.org/314
Reviewed-by: Anthony Barbier <Anthony.barbier@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core')
-rw-r--r-- | src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp | 12 |
1 files changed, 2 insertions, 10 deletions
diff --git a/src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp b/src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp index d3ab5490a4..09836f1d6b 100644 --- a/src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp +++ b/src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp @@ -431,21 +431,13 @@ void output_stage_nhwc<int32_t, uint8_t, false, false>(ITensor *input, const ITe uint8x16_t min = vdupq_n_u8(0); uint8x16_t max = vdupq_n_u8(255); - Window window_bias = window; - window_bias.set(Window::DimY, Window::Dimension(0, 0, 0)); - window_bias.set(Window::DimZ, Window::Dimension(0, 0, 0)); - window_bias.set(3, Window::Dimension(0, 0, 0)); - Iterator in(input, window); - Iterator bi(bias, window_bias); - Iterator out(output, window); execute_window_loop(window, [&](const Coordinates & id) { - // Get bias and pointer to input + // Get pointer to input const auto in_ptr = reinterpret_cast<int32_t *>(in.ptr()); - // Accumulate bias int32x4x4_t v_in = { { @@ -459,7 +451,7 @@ void output_stage_nhwc<int32_t, uint8_t, false, false>(ITensor *input, const ITe const auto out_ptr = out.ptr(); vst1q_u8(out_ptr, finalize_quantization<false>(v_in, result_fixedpoint_multiplier, result_shift, result_offset_after_shift_s32, min, max)); }, - in, bi, out); + in, out); } } // namespace |