diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-10-27 18:07:04 +0000 |
|---|---|---|
| committer | Jim Flynn <jim.flynn@arm.com> | 2020-10-28 11:59:04 +0000 |
| commit | 0103e769491e1ee4474c55ead522468db4b3ebfb (patch) | |
| tree | 25d581c3d4f765ce321f9d77c342b3cff3e86dff | |
| parent | 0534e0364473c0b1244f96462cbde1808e92ce81 (diff) | |
| download | armnn-0103e769491e1ee4474c55ead522468db4b3ebfb.tar.gz | |
Fix BackendHint missing when cloning a layer
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I550fb59469af39f8fa3415843e973f06b18485e5
| -rw-r--r-- | src/armnn/layers/LayerCloneBase.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
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<LayerType>(std::forward<Params>(params)...); + layer->BackendSelectionHint(GetBackendHint()); layer->SetBackendId(GetBackendId()); layer->SetGuid(GetGuid()); layer->SetShapeInferenceMethod(m_ShapeInferenceMethod); |