diff options
Diffstat (limited to 'src/armnn/Network.hpp')
-rw-r--r-- | src/armnn/Network.hpp | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp index b90e3c2f8d..8db968a3f9 100644 --- a/src/armnn/Network.hpp +++ b/src/armnn/Network.hpp @@ -13,6 +13,7 @@ #include <string> #include <vector> +#include <map> #include <memory> #include "Layer.hpp" @@ -229,4 +230,27 @@ private: std::unique_ptr<Graph> m_Graph; }; + + +struct OptimizationResult +{ + bool m_Warning; + bool m_Error; + + OptimizationResult() + : m_Warning(false) + , m_Error(false) + {} +}; + +using BackendsMap = std::map<BackendId, std::unique_ptr<class IBackendInternal>>; + +BackendsMap CreateSupportedBackends(TensorHandleFactoryRegistry& handleFactoryRegistry, + struct BackendSettings& backendSettings); + +OptimizationResult SelectTensorHandleStrategy(Graph& optGraph, + BackendsMap& backends, + TensorHandleFactoryRegistry& registry, + Optional<std::vector<std::string>&> errMessages); + } // namespace armnn |