diff options
| -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; |