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

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 65ae14f..87d2921 100644 (file)
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -968,6 +968,9 @@ bool RefLayerSupport::IsFullyConnectedSupported(const TensorInfo& input,
         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;