aboutsummaryrefslogtreecommitdiff
path: root/src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2018-11-29 13:58:31 +0000
committerGeorgios Pinitas <georgios.pinitas@arm.com>2018-11-29 16:36:56 +0000
commit551852fb43a5cf7ed9ac8679517b103e9af8a38f (patch)
tree6d2d3fc7349c9153101684722e520d5fa99d5eb6 /src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp
parent618493d9823936799501334d06572c5f2d8da319 (diff)
downloadComputeLibrary-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/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp')
-rw-r--r--src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp12
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