diff options
| author | Moritz Pflanzer <moritz.pflanzer@arm.com> | 2017-08-09 11:43:18 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | 484e7b3724c0e77751b5bed05180271fd5376e5d (patch) | |
| tree | 1ff3fa4bd80b1aef91086ace9b0c8f006445a217 /src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp | |
| parent | 768e9f180f4082a713a2eb9f3b98911c767d570b (diff) | |
| download | ComputeLibrary-484e7b3724c0e77751b5bed05180271fd5376e5d.tar.gz | |
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>
Diffstat (limited to 'src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp')
| -rw-r--r-- | src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp b/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp index a4fc494f16..6ed3791ce5 100644 --- a/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp +++ b/src/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.cpp @@ -48,7 +48,7 @@ void NEGEMMMatrixAccumulateBiasesKernel::configure(ITensor *accum, const ITensor 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; |