aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Runtime.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/Runtime.hpp')
-rw-r--r--src/armnn/Runtime.hpp12
1 files changed, 6 insertions, 6 deletions
diff --git a/src/armnn/Runtime.hpp b/src/armnn/Runtime.hpp
index 29bb6808d8..2679e3cc98 100644
--- a/src/armnn/Runtime.hpp
+++ b/src/armnn/Runtime.hpp
@@ -9,7 +9,8 @@
#include <armnn/INetwork.hpp>
#include <armnn/IRuntime.hpp>
#include <armnn/Tensor.hpp>
-#include <backends/cl/ClContextControl.hpp>
+#include <armnn/BackendId.hpp>
+#include <backends/IBackendContext.hpp>
#include <mutex>
#include <unordered_map>
@@ -85,14 +86,13 @@ private:
}
mutable std::mutex m_Mutex;
-
std::unordered_map<NetworkId, std::unique_ptr<LoadedNetwork>> m_LoadedNetworks;
-
- ClContextControl m_ClContextControl;
-
+ CreationOptions m_Options;
int m_NetworkIdCounter;
-
DeviceSpec m_DeviceSpec;
+
+ using BackendContextMap = std::unordered_map<BackendId, IBackendContextUniquePtr>;
+ BackendContextMap m_BackendContexts;
};
}