aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/LayerCloneBase.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/LayerCloneBase.hpp')
-rw-r--r--src/armnn/layers/LayerCloneBase.hpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/layers/LayerCloneBase.hpp b/src/armnn/layers/LayerCloneBase.hpp
index 3671d6642a..a8ff52b86a 100644
--- a/src/armnn/layers/LayerCloneBase.hpp
+++ b/src/armnn/layers/LayerCloneBase.hpp
@@ -1,5 +1,5 @@
//
-// Copyright © 2017 Arm Ltd. All rights reserved.
+// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
@@ -17,6 +17,7 @@ LayerType* Layer::CloneBase(Graph& graph, Params&& ... params) const
layer->SetBackendId(GetBackendId());
layer->SetGuid(GetGuid());
+ layer->SetShapeInferenceMethod(m_ShapeInferenceMethod);
return layer;
}