diff options
Diffstat (limited to 'src/armnn/Runtime.hpp')
-rw-r--r-- | src/armnn/Runtime.hpp | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/src/armnn/Runtime.hpp b/src/armnn/Runtime.hpp index 694e1e5f9a..e4d4d4ddb9 100644 --- a/src/armnn/Runtime.hpp +++ b/src/armnn/Runtime.hpp @@ -10,7 +10,8 @@ #include <armnn/IRuntime.hpp> #include <armnn/Tensor.hpp> #include <armnn/BackendId.hpp> -#include <backendsCommon/IBackendContext.hpp> + +#include <cl/ClContextControl.hpp> #include <mutex> #include <unordered_map> @@ -86,13 +87,14 @@ private: } mutable std::mutex m_Mutex; + std::unordered_map<NetworkId, std::unique_ptr<LoadedNetwork>> m_LoadedNetworks; - CreationOptions m_Options; + + ClContextControl m_ClContextControl; + int m_NetworkIdCounter; - DeviceSpec m_DeviceSpec; - using BackendContextMap = std::unordered_map<BackendId, IBackendContextUniquePtr>; - BackendContextMap m_BackendContexts; + DeviceSpec m_DeviceSpec; }; } |