diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-04-29 12:58:17 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-04-29 18:26:54 +0100 |
commit | 366d72306fb62e2cf5ebd7595e9692c667374566 (patch) | |
tree | 2ac89a81744f4d60c0f6027707d28317f0a4674f /src/backends/reference | |
parent | ce691bbd64d1a3c4f61bc3348bae1e4f6a3b7077 (diff) | |
download | armnn-366d72306fb62e2cf5ebd7595e9692c667374566.tar.gz |
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
Diffstat (limited to 'src/backends/reference')
-rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
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; |