diff options
| author | Tee Jung <tee.ty.jung@openedges.com> | 2019-11-01 11:55:21 +0000 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-04 09:07:50 +0000 |
| commit | d94efa8a7c6476fb50b3434723fda22859c236ad (patch) | |
| tree | 66fd714bf2390f4dceaf6beefc75702ec9c49d02 | |
| parent | 584a2b83338a1190eb2d1840a477fd65b7dc4844 (diff) | |
| download | armnn-d94efa8a7c6476fb50b3434723fda22859c236ad.tar.gz | |
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
| -rw-r--r-- | src/armnnOnnxParser/test/FullyConnected.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
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 @@ -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" |