From 366d72306fb62e2cf5ebd7595e9692c667374566 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Wed, 29 Apr 2020 12:58:17 +0100 Subject: IVGCVSW-4686 Fix NNT GeneratedTests.Tests * Add bias dimension check for FullyConnected layer support Signed-off-by: Narumol Prangnawarat Change-Id: I66db5332069a606b5d1f6c10faee38760b26a8f8 --- src/backends/reference/RefLayerSupport.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 65ae14ff40..87d2921952 100644 --- 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; -- cgit v1.2.1