From 4422ceca976a88aac49b21808a43e465bc87a35e Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Mon, 22 Mar 2021 17:51:06 +0000 Subject: Fix graph copy memory spike * Change layer storage of ConstTensors to std::shared_ptr * Change clone to share ConstTensor rather than copy * Remove uses of non-const GetTensor() call * Reduce scope of non-optimized network in ExeNet, so memory can be released after use Signed-off-by: Finn Williams Change-Id: Ibb2c7309d12411d21405bd6024c76bcdf5404545 --- src/armnn/layers/FullyConnectedLayer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/armnn/layers/FullyConnectedLayer.cpp') diff --git a/src/armnn/layers/FullyConnectedLayer.cpp b/src/armnn/layers/FullyConnectedLayer.cpp index 6d0b57a84c..44c8920136 100644 --- a/src/armnn/layers/FullyConnectedLayer.cpp +++ b/src/armnn/layers/FullyConnectedLayer.cpp @@ -44,11 +44,11 @@ FullyConnectedLayer* FullyConnectedLayer::Clone(Graph& graph) const auto layer = CloneBase(graph, m_Param, GetName()); if (m_Param.m_ConstantWeights) { - layer->m_Weight = m_Weight ? std::make_unique(*m_Weight) : nullptr; + layer->m_Weight = m_Weight ? m_Weight : nullptr; if (layer->m_Param.m_BiasEnabled) { - layer->m_Bias = m_Bias ? std::make_unique(*m_Bias) : nullptr; + layer->m_Bias = m_Bias ? m_Bias : nullptr; } } return std::move(layer); -- cgit v1.2.1