Match shape in between initializer and tensor

 * Added extra dims definition for the second dimension of the weights

Signed-off-by: Jung Tae-young tee.ty.jung@openedges.com
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Change-Id: Ifeeff5c3d4733dbb9c246242fc4b796eb4d515a5

diff --git a/src/armnnOnnxParser/test/FullyConnected.cpp b/src/armnnOnnxParser/test/FullyConnected.cpp
index 395ff90..133b8a6 100644 (file)
--- a/src/armnnOnnxParser/test/FullyConnected.cpp
+++ b/src/armnnOnnxParser/test/FullyConnected.cpp
@@ -56,6 +56,7 @@ struct MatMulFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxParser:
                         }
                         initializer {
                           dims: 1
+                          dims: 1
                           data_type: 1
                           float_data: 17.0
                           name: "Const"
@@ -145,6 +146,7 @@ struct FullyConnectedFixture : public armnnUtils::ParserPrototxtFixture<armnnOnn
                         }
                         initializer {
                           dims: 1
+                          dims: 1
                           data_type: 1
                           float_data: 2
                           name: "Weight"
@@ -284,6 +286,7 @@ struct MatMulUsedInTwoFcFixture : public armnnUtils::ParserPrototxtFixture<armnn
                         }
                         initializer {
                           dims: 1
+                          dims: 1
                           data_type: 1
                           float_data: 2
                           name: "Weight"
@@ -488,6 +491,7 @@ struct MatMulUsedInTwoFcStaggeredFixture : public armnnUtils::ParserPrototxtFixt
                         }
                         initializer {
                           dims: 1
+                          dims: 1
                           data_type: 1
                           float_data: 2
                           name: "Weight"