IVGCVSW-4314 Per-layer backend hint API

Change-Id: I6ddcffe792e39b17fcdb8af7f13f4a689ef8019d
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>

1 files changed, 8 insertions, 0 deletions

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;