diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-02-11 13:57:07 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-02-11 13:57:07 +0000 |
commit | 49bdb794170c3d25e3e51fc7b4c267c3d8dbcebf (patch) | |
tree | 1dcfc4495d48320e27354e7a2ff85b6aa3ada5ea /src/armnnTfLiteParser/test/Reduce.cpp | |
parent | a35b40bb58b2de20893f64a5941ccc54d64a8ffa (diff) | |
download | armnn-49bdb794170c3d25e3e51fc7b4c267c3d8dbcebf.tar.gz |
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 <sadik.armagan@arm.com>
Change-Id: I98b6e00ccba1d8ea8c845cb1ae28840e42339629
Diffstat (limited to 'src/armnnTfLiteParser/test/Reduce.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/Reduce.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
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) |