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/ConstantLayer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/armnn/layers/ConstantLayer.cpp') diff --git a/src/armnn/layers/ConstantLayer.cpp b/src/armnn/layers/ConstantLayer.cpp index 31e9e974cf..8ae34b6709 100644 --- a/src/armnn/layers/ConstantLayer.cpp +++ b/src/armnn/layers/ConstantLayer.cpp @@ -32,7 +32,7 @@ ConstantLayer* ConstantLayer::Clone(Graph& graph) const // Cloned layers share the same layer output object. auto layer = CloneBase(graph, GetName()); - layer->m_LayerOutput = m_LayerOutput ? std::make_unique(*m_LayerOutput) : nullptr; + layer->m_LayerOutput = m_LayerOutput ? m_LayerOutput : nullptr; return std::move(layer); } -- cgit v1.2.1