diff options
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 |