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 --- include/armnn/Descriptors.hpp | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) (limited to 'include/armnn') 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. -- cgit v1.2.1