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/Network.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/Network.hpp')
-rw-r--r-- | src/armnn/Network.hpp | 20 |
1 files changed, 18 insertions, 2 deletions
diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp index 5da681306c..089b46c9ca 100644 --- a/src/armnn/Network.hpp +++ b/src/armnn/Network.hpp @@ -17,6 +17,7 @@ #include <map> #include <memory> +#include "Graph.hpp" #include "Layer.hpp" namespace armnn @@ -286,10 +287,19 @@ struct OptimizationResult bool m_Warning; bool m_Error; + OptimizationResult(bool warning, bool error) + : m_Warning(warning) + , m_Error(error) + {} + OptimizationResult() - : m_Warning(false) - , m_Error(false) + : OptimizationResult(false, false) {} + + bool IsOk() const { return !m_Warning && !m_Error; } + bool IsWarningOnly() const { return m_Warning && !m_Error; } + bool IsError() const { return m_Error; } + }; using BackendsMap = std::map<BackendId, std::unique_ptr<class IBackendInternal>>; @@ -302,4 +312,10 @@ OptimizationResult SelectTensorHandleStrategy(Graph& optGraph, TensorHandleFactoryRegistry& registry, Optional<std::vector<std::string>&> errMessages); +OptimizationResult AssignBackends(OptimizedNetwork* optNetObjPtr, + BackendSettings& backendSettings, + Graph::Iterator& firstLayer, + Graph::Iterator& lastLayer, + Optional<std::vector<std::string>&> errMessages); + } // namespace armnn |