diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2017-11-29 11:56:49 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:41:39 +0000 |
commit | 4db041ec8adc565452b528fd36f5534999ccdfd0 (patch) | |
tree | 3cd16c43f81d1d98f7b006912253f8850209f6fb /src/core/NEON/kernels/NEGEMMMatrixVectorMultiplyKernel.cpp | |
parent | 5420b28d54c5238ed4c4aa2ccf02a7c3855c6760 (diff) | |
download | ComputeLibrary-4db041ec8adc565452b528fd36f5534999ccdfd0.tar.gz |
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 <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NEGEMMMatrixVectorMultiplyKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NEGEMMMatrixVectorMultiplyKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
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); |