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 /include/armnn | |
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 'include/armnn')
-rw-r--r-- | include/armnn/Descriptors.hpp | 12 |
1 files changed, 2 insertions, 10 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index d6e37e535e..a8e68aa8c1 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -1290,26 +1290,18 @@ struct LogicalBinaryDescriptor struct ReduceDescriptor { ReduceDescriptor() - : m_TargetHeight(0) - , m_TargetWidth(0) - , m_KeepDims(false) + : m_KeepDims(false) , m_vAxis() , m_ReduceOperation(ReduceOperation::Sum) {} bool operator ==(const ReduceDescriptor& rhs) const { - return m_TargetHeight == rhs.m_TargetHeight && - m_TargetWidth == rhs.m_TargetWidth && - m_KeepDims == rhs.m_KeepDims && + return m_KeepDims == rhs.m_KeepDims && m_vAxis == rhs.m_vAxis && m_ReduceOperation == rhs.m_ReduceOperation; } - /// Target height value. - uint32_t m_TargetHeight; - /// Target width value. - uint32_t m_TargetWidth; /// if true then output shape has no change. bool m_KeepDims; /// The indices of the dimensions to reduce. |