Fix BackendHint missing when cloning a layer
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>
Tue, 27 Oct 2020 18:07:04 +0000 (18:07 +0000)
committerJim Flynn <jim.flynn@arm.com>
Wed, 28 Oct 2020 11:59:04 +0000 (11:59 +0000)
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I550fb59469af39f8fa3415843e973f06b18485e5

src/armnn/layers/LayerCloneBase.hpp

index a8ff52b..348b1f3 100644 (file)
@@ -15,6 +15,7 @@ LayerType* Layer::CloneBase(Graph& graph, Params&& ... params) const
 {
     LayerType* const layer = graph.AddLayer<LayerType>(std::forward<Params>(params)...);
 
+    layer->BackendSelectionHint(GetBackendHint());
     layer->SetBackendId(GetBackendId());
     layer->SetGuid(GetGuid());
     layer->SetShapeInferenceMethod(m_ShapeInferenceMethod);