From 49bdb794170c3d25e3e51fc7b4c267c3d8dbcebf Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Thu, 11 Feb 2021 13:57:07 +0000 Subject: MLCE-360 'ReduceLayer InferOutputShape Issue' * Updated ParseReduce() function in TfLiteParser to read correct axis data * Remove unused m_TargetWidth and m_TargetHight from ReduceDescriptor * Updated the ArmNN Serializer Schema Signed-off-by: Sadik Armagan Change-Id: I98b6e00ccba1d8ea8c845cb1ae28840e42339629 --- src/armnnTfLiteParser/test/Reduce.cpp | 4 ++-- src/armnnTfLiteParser/test/Sum.cpp | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) (limited to 'src/armnnTfLiteParser/test') diff --git a/src/armnnTfLiteParser/test/Reduce.cpp b/src/armnnTfLiteParser/test/Reduce.cpp index 622d54e8b5..c2a22f0b86 100644 --- a/src/armnnTfLiteParser/test/Reduce.cpp +++ b/src/armnnTfLiteParser/test/Reduce.cpp @@ -90,7 +90,7 @@ struct ReduceMaxFixture : public ParserFlatbuffersFixture struct SimpleReduceMaxFixture : public ReduceMaxFixture { - SimpleReduceMaxFixture() : ReduceMaxFixture("[ 1, 1, 2, 3 ]", "[ 1, 1, 1, 3 ]", "[ 1 ]", "[ 2 ]") {} + SimpleReduceMaxFixture() : ReduceMaxFixture("[ 1, 1, 2, 3 ]", "[ 1, 1, 1, 3 ]", "[ 1 ]", "[ 2,0,0,0 ]") {} }; BOOST_FIXTURE_TEST_CASE(ParseReduceMax, SimpleReduceMaxFixture) @@ -179,7 +179,7 @@ struct ReduceMinFixture : public ParserFlatbuffersFixture struct SimpleReduceMinFixture : public ReduceMinFixture { - SimpleReduceMinFixture() : ReduceMinFixture("[ 1, 1, 2, 3 ]", "[ 1, 1, 1, 3 ]", "[ 1 ]", "[ 2 ]") {} + SimpleReduceMinFixture() : ReduceMinFixture("[ 1, 1, 2, 3 ]", "[ 1, 1, 1, 3 ]", "[ 1 ]", "[ 2, 0, 0, 0 ]") {} }; BOOST_FIXTURE_TEST_CASE(ParseReduceMin, SimpleReduceMinFixture) diff --git a/src/armnnTfLiteParser/test/Sum.cpp b/src/armnnTfLiteParser/test/Sum.cpp index 22b19ae058..177bcd52de 100644 --- a/src/armnnTfLiteParser/test/Sum.cpp +++ b/src/armnnTfLiteParser/test/Sum.cpp @@ -90,7 +90,7 @@ struct SumFixture : public ParserFlatbuffersFixture struct SimpleSumFixture : public SumFixture { - SimpleSumFixture() : SumFixture("[ 1, 3, 2, 4 ]", "[ 1, 1, 1, 4 ]", "[ 2 ]", "[ 1, 2 ]") {} + SimpleSumFixture() : SumFixture("[ 1, 3, 2, 4 ]", "[ 1, 1, 1, 4 ]", "[ 2 ]", "[ 1, 0, 0, 0, 2, 0, 0, 0 ]") {} }; BOOST_FIXTURE_TEST_CASE(ParseSum, SimpleSumFixture) -- cgit v1.2.1