diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2021-03-12 09:15:35 +0000 |
|---|---|---|
| committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-03-12 09:20:33 +0000 |
| commit | 58fb0ee76104ee281efbace28f1edb0a764d5044 (patch) | |
| tree | 7993d1a379515864577603993a4c795fe3f74a63 | |
| parent | 66dbf5b9a3f1bf9d96c0dcaff824b159d1b072a9 (diff) | |
| download | armnn-58fb0ee76104ee281efbace28f1edb0a764d5044.tar.gz | |
IVGCVSW-5767 'Regression on SqueezeNet v1.1'
* Update ReduceLayer to use descriptor parameters
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I2ff240f63534de0fd647428cf411763829412443
| -rw-r--r-- | src/armnn/layers/ReduceLayer.cpp | 12 |
1 files changed, 11 insertions, 1 deletions
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<IWorkload> 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<ReduceLayer>(graph, m_Param, GetName()); + auto layer = CloneBase<ReduceLayer>(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() |