commit | 484e7b3724c0e77751b5bed05180271fd5376e5d | [log] [tgz] |
---|---|---|
author | Moritz Pflanzer <moritz.pflanzer@arm.com> | Wed Aug 09 11:43:18 2017 +0100 |
committer | Anthony Barbier <anthony.barbier@arm.com> | Fri Nov 02 16:35:24 2018 +0000 |
tree | 1ff3fa4bd80b1aef91086ace9b0c8f006445a217 | |
parent | 768e9f180f4082a713a2eb9f3b98911c767d570b [diff] [blame] |
COMPMID-417: Cleanup NEON FullyConnectedLayer Change-Id: Ie02a0a1a28ca2771e29a5e6552242caf0f6db1cf Reviewed-on: http://mpd-gerrit.cambridge.arm.com/83555 Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com> Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp b/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp index a4fc494..6ed3791 100644 --- a/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp +++ b/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp
@@ -48,7 +48,7 @@ ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(accum, 1, DataType::QS8, DataType::QS16, DataType::F16, DataType::F32); ARM_COMPUTE_ERROR_ON_MISMATCHING_DATA_TYPES(biases, accum); ARM_COMPUTE_ERROR_ON_MISMATCHING_FIXED_POINT_POSITION(biases, accum); - ARM_COMPUTE_ERROR_ON(biases->info()->num_dimensions() != 1); + ARM_COMPUTE_ERROR_ON(biases->info()->num_dimensions() > 1); _biases = biases; _accum = accum;