aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/DynamicBackend.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/DynamicBackend.hpp')
-rw-r--r--src/backends/backendsCommon/DynamicBackend.hpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/DynamicBackend.hpp b/src/backends/backendsCommon/DynamicBackend.hpp
index b202309c31..34c001958d 100644
--- a/src/backends/backendsCommon/DynamicBackend.hpp
+++ b/src/backends/backendsCommon/DynamicBackend.hpp
@@ -6,6 +6,7 @@
#pragma once
#include "IBackendInternal.hpp"
+#include "BackendRegistry.hpp"
#include <functional>
#include <memory>
@@ -13,7 +14,7 @@
namespace armnn
{
-class DynamicBackend
+class DynamicBackend final
{
public:
using HandleCloser = std::function<void(const void*)>;
@@ -25,10 +26,13 @@ public:
BackendId GetBackendId();
BackendVersion GetBackendVersion();
IBackendInternalUniquePtr GetBackend();
+ BackendRegistry::FactoryFunction GetFactoryFunction();
private:
+ /// Private utility functions
template<typename BackendFunctionType>
BackendFunctionType SetFunctionPointer(const std::string& backendFunctionName);
+ IBackendInternalUniquePtr CreateBackend();
/// Backend function pointer types
using IdFunctionType = const char*(*)();