diff options
author | FinnWilliamsArm <Finn.Williams@Arm.com> | 2020-01-08 14:57:47 +0000 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-01-15 14:58:23 +0000 |
commit | 42defa6c2bc2574f39963554937bdf81e49be449 (patch) | |
tree | 1240c2b1fb3dff143db0997562dc987f834ed03b /ConversionUtils.hpp | |
parent | 2c90f75496d2c0210a43f35ddb75f6492dd06490 (diff) | |
download | android-nn-driver-branches/android-nn-driver_19_11.tar.gz |
IVGCVSW-4315 Fix Fully Connected infer output shape bugv19.11.1branches/android-nn-driver_19_11
Change-Id: If4fd1abdedf7de2046435d418fb1ee95ceb73419
Signed-off-by: FinnWilliamsArm <Finn.Williams@Arm.com>
Diffstat (limited to 'ConversionUtils.hpp')
-rw-r--r-- | ConversionUtils.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp index a342d399..3a144f78 100644 --- a/ConversionUtils.hpp +++ b/ConversionUtils.hpp @@ -2645,6 +2645,14 @@ bool ConvertFullyConnected(const Operation& operation, const Model& model, Conve desc.m_TransposeWeightMatrix = true; desc.m_BiasEnabled = true; + if (!VerifyFullyConnectedShapes(reshapedInfo.GetShape(), + weights.GetInfo().GetShape(), + outputInfo.GetShape(), + desc.m_TransposeWeightMatrix)) + { + return Fail("%s: Expected outputShape does not match actual outputShape", __func__); + } + bool isSupported = false; FORWARD_LAYER_SUPPORT_FUNC(__func__, IsFullyConnectedSupported, |