diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2020-01-03 16:53:38 +0000 |
---|---|---|
committer | Derek Lamberti <derek.lamberti@arm.com> | 2020-03-03 14:56:00 +0000 |
commit | 4a9e24bfc51eec7e593470091fb7e6e435ae3991 (patch) | |
tree | a9be306fe10d6cd7d07fb06817c224e5e5eb8ada /src/armnn/Layer.hpp | |
parent | 9077473e0db8548506360a5196a1514f785332bb (diff) | |
download | armnn-4a9e24bfc51eec7e593470091fb7e6e435ae3991.tar.gz |
IVGCVSW-4314 Per-layer backend hint API
Change-Id: I6ddcffe792e39b17fcdb8af7f13f4a689ef8019d
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'src/armnn/Layer.hpp')
-rw-r--r-- | src/armnn/Layer.hpp | 8 |
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; |