From 0103e769491e1ee4474c55ead522468db4b3ebfb Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Tue, 27 Oct 2020 18:07:04 +0000 Subject: Fix BackendHint missing when cloning a layer Signed-off-by: Narumol Prangnawarat Change-Id: I550fb59469af39f8fa3415843e973f06b18485e5 --- src/armnn/layers/LayerCloneBase.hpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/armnn/layers/LayerCloneBase.hpp b/src/armnn/layers/LayerCloneBase.hpp index a8ff52b86a..348b1f3bf6 100644 --- a/src/armnn/layers/LayerCloneBase.hpp +++ b/src/armnn/layers/LayerCloneBase.hpp @@ -15,6 +15,7 @@ LayerType* Layer::CloneBase(Graph& graph, Params&& ... params) const { LayerType* const layer = graph.AddLayer(std::forward(params)...); + layer->BackendSelectionHint(GetBackendHint()); layer->SetBackendId(GetBackendId()); layer->SetGuid(GetGuid()); layer->SetShapeInferenceMethod(m_ShapeInferenceMethod); -- cgit v1.2.1