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/armnnSerializer/Serializer.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/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index a2217a3dc4..28afac7b62 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -911,8 +911,6 @@ void SerializerVisitor::VisitReduceLayer(const armnn::IConnectableLayer* layer, { auto fbReduceBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Reduce); auto fbDescriptor = CreateReduceDescriptor(m_flatBufferBuilder, - reduceDescriptor.m_TargetHeight, - reduceDescriptor.m_TargetWidth, reduceDescriptor.m_KeepDims, m_flatBufferBuilder.CreateVector(reduceDescriptor.m_vAxis), GetFlatBufferReduceOperation(reduceDescriptor.m_ReduceOperation)); |