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/armnnDeserializer | |
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/armnnDeserializer')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index e98ff15aa9..cbc4758e0e 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -2118,8 +2118,6 @@ void IDeserializer::DeserializerImpl::ParseReduce(GraphPtr graph, unsigned int l auto flatBufferAxis = fbDescriptor->axis(); armnn::ReduceDescriptor descriptor; - descriptor.m_TargetHeight = fbDescriptor->targetHeight(); - descriptor.m_TargetWidth = fbDescriptor->targetWidth(); descriptor.m_KeepDims = fbDescriptor->keepDims(); descriptor.m_vAxis = std::vector<unsigned int>(flatBufferAxis->begin(), flatBufferAxis->end()); descriptor.m_ReduceOperation = ToReduceOperation(fbDescriptor->reduceOperation()); |