From 4a9e24bfc51eec7e593470091fb7e6e435ae3991 Mon Sep 17 00:00:00 2001
From: Derek Lamberti <derek.lamberti@arm.com>
Date: Fri, 3 Jan 2020 16:53:38 +0000
Subject: IVGCVSW-4314 Per-layer backend hint API

Change-Id: I6ddcffe792e39b17fcdb8af7f13f4a689ef8019d
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
---
 src/armnn/Layer.hpp | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/armnn/Layer.hpp')

diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp
index c62b67ac0b..5ad38f0b9e 100644
--- a/src/armnn/Layer.hpp
+++ b/src/armnn/Layer.hpp
@@ -320,6 +320,13 @@ public:
     const std::list<std::string>& GetRelatedLayerNames() { return m_RelatedLayerNames; }
 
     virtual void Reparent(Graph& dest, std::list<Layer*>::const_iterator iterator) = 0;
+
+    void BackendSelectionHint(Optional<BackendId> backend) final
+    {
+        m_BackendHint = backend;
+    }
+    Optional<BackendId> GetBackendHint() const { return m_BackendHint; }
+
 protected:
     // Graph needs access to the virtual destructor.
     friend class Graph;
@@ -371,6 +378,7 @@ private:
 
     const LayerType m_Type;
     BackendId m_BackendId;
+    Optional<BackendId> m_BackendHint;
 
     /// Used for sorting.
     mutable LayerPriority m_Priority = 0;
-- 
cgit v1.2.1