From 4db041ec8adc565452b528fd36f5534999ccdfd0 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Wed, 29 Nov 2017 11:56:49 +0000 Subject: COMPMID-701 Fixed std::bad_alloc in NEON DepthwiseConvolutionLayer Change-Id: I249749c38a5f96c0d1fdbc217acb435307627499 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/111074 Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com Reviewed-by: Georgios Pinitas Reviewed-by: Anthony Barbier --- src/core/NEON/kernels/NEGEMMMatrixVectorMultiplyKernel.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/core/NEON/kernels/NEGEMMMatrixVectorMultiplyKernel.cpp') diff --git a/src/core/NEON/kernels/NEGEMMMatrixVectorMultiplyKernel.cpp b/src/core/NEON/kernels/NEGEMMMatrixVectorMultiplyKernel.cpp index c28dcf7463..fe79df2528 100644 --- a/src/core/NEON/kernels/NEGEMMMatrixVectorMultiplyKernel.cpp +++ b/src/core/NEON/kernels/NEGEMMMatrixVectorMultiplyKernel.cpp @@ -62,7 +62,7 @@ void NEGEMMMatrixVectorMultiplyKernel::configure(const ITensor *input0, const IT AccessWindowHorizontal input0_access(input0->info(), 0, num_elems_read_per_iteration); AccessWindowHorizontal input1_access(input1->info(), 0, num_elems_read_per_iteration); - AccessWindowHorizontal output_access(output->info(), 0, 1); + AccessWindowStatic output_access(output->info(), 0, 0, output->info()->dimension(0), output->info()->dimension(1)); update_window_and_padding(win, input0_access, input1_access, output_access); -- cgit v1.2.1