From 58fb0ee76104ee281efbace28f1edb0a764d5044 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Fri, 12 Mar 2021 09:15:35 +0000 Subject: IVGCVSW-5767 'Regression on SqueezeNet v1.1' * Update ReduceLayer to use descriptor parameters Signed-off-by: Sadik Armagan Change-Id: I2ff240f63534de0fd647428cf411763829412443 --- src/armnn/layers/ReduceLayer.cpp | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/src/armnn/layers/ReduceLayer.cpp b/src/armnn/layers/ReduceLayer.cpp index b68cd2eabc..31a2dfa479 100644 --- a/src/armnn/layers/ReduceLayer.cpp +++ b/src/armnn/layers/ReduceLayer.cpp @@ -22,12 +22,22 @@ ReduceLayer::ReduceLayer(const ReduceDescriptor& param, const char* name) std::unique_ptr ReduceLayer::CreateWorkload(const IWorkloadFactory& factory) const { ReduceQueueDescriptor descriptor; + descriptor.m_Parameters.m_vAxis = m_Param.m_vAxis; + descriptor.m_Parameters.m_KeepDims = m_Param.m_KeepDims; + descriptor.m_Parameters.m_ReduceOperation = m_Param.m_ReduceOperation; + SetAdditionalInfo(descriptor); + return factory.CreateReduce(descriptor, PrepInfoAndDesc(descriptor)); } ReduceLayer* ReduceLayer::Clone(Graph& graph) const { - return CloneBase(graph, m_Param, GetName()); + auto layer = CloneBase(graph, m_Param, GetName()); + layer->m_Param.m_vAxis = m_Param.m_vAxis; + layer->m_Param.m_KeepDims = m_Param.m_KeepDims; + layer->m_Param.m_ReduceOperation = m_Param.m_ReduceOperation; + + return std::move(layer); } void ReduceLayer::ValidateTensorShapesFromInputs() -- cgit v1.2.1