From d94efa8a7c6476fb50b3434723fda22859c236ad Mon Sep 17 00:00:00 2001
From: Tee Jung <tee.ty.jung@openedges.com>
Date: Fri, 1 Nov 2019 11:55:21 +0000
Subject: 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
---
 src/armnnOnnxParser/test/FullyConnected.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/armnnOnnxParser/test/FullyConnected.cpp b/src/armnnOnnxParser/test/FullyConnected.cpp
index 395ff90bb9..133b8a6848 100644
--- a/src/armnnOnnxParser/test/FullyConnected.cpp
+++ b/src/armnnOnnxParser/test/FullyConnected.cpp
@@ -55,6 +55,7 @@ struct MatMulFixture : public armnnUtils::ParserPrototxtFixture<armnnOnnxParser:
                           }
                         }
                         initializer {
+                          dims: 1
                           dims: 1
                           data_type: 1
                           float_data: 17.0
@@ -144,6 +145,7 @@ struct FullyConnectedFixture : public armnnUtils::ParserPrototxtFixture<armnnOnn
                           }
                         }
                         initializer {
+                          dims: 1
                           dims: 1
                           data_type: 1
                           float_data: 2
@@ -283,6 +285,7 @@ struct MatMulUsedInTwoFcFixture : public armnnUtils::ParserPrototxtFixture<armnn
                           }
                         }
                         initializer {
+                          dims: 1
                           dims: 1
                           data_type: 1
                           float_data: 2
@@ -487,6 +490,7 @@ struct MatMulUsedInTwoFcStaggeredFixture : public armnnUtils::ParserPrototxtFixt
                           }
                         }
                         initializer {
+                          dims: 1
                           dims: 1
                           data_type: 1
                           float_data: 2
-- 
cgit v1.2.1