commit 366d72306fb62e2cf5ebd7595e9692c667374566
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Wed Apr 29 12:58:17 2020 +0100
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Wed Apr 29 18:26:54 2020 +0100
tree 2ac89a81744f4d60c0f6027707d28317f0a4674f
parent ce691bbd64d1a3c4f61bc3348bae1e4f6a3b7077

IVGCVSW-4686 Fix NNT GeneratedTests.Tests

 * Add bias dimension check for FullyConnected layer support

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I66db5332069a606b5d1f6c10faee38760b26a8f8

src/backends/reference/RefLayerSupport.cpp

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 65ae14f..87d2921 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -968,6 +968,9 @@
         supported &= CheckSupportRule(BiasAndWeightsTypesCompatible(weights, supportedBiasTypes), reasonIfUnsupported,
                                       "Reference Fully Connected: bias type inferred from weights is incompatible.");
 
+        supported &= CheckSupportRule(TensorNumDimensionsAreCorrect(biases, 1U), reasonIfUnsupported,
+                                      "Reference Fully Connected: bias must have 1 dimension.");
+
     }
 
     return supported;