diff options
Diffstat (limited to 'src/armnn/LoadedNetwork.hpp')
-rw-r--r-- | src/armnn/LoadedNetwork.hpp | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/src/armnn/LoadedNetwork.hpp b/src/armnn/LoadedNetwork.hpp index 3deb8bc2e2..51eb04f3df 100644 --- a/src/armnn/LoadedNetwork.hpp +++ b/src/armnn/LoadedNetwork.hpp @@ -11,13 +11,12 @@ #include "LayerFwd.hpp" #include "Profiling.hpp" -#include <backends/reference/RefWorkloadFactory.hpp> -#include <backends/neon/NeonWorkloadFactory.hpp> -#include <backends/cl/ClWorkloadFactory.hpp> +#include <backends/IBackendInternal.hpp> #include <backends/Workload.hpp> #include <backends/WorkloadFactory.hpp> #include <mutex> +#include <unordered_map> namespace cl { @@ -62,9 +61,11 @@ private: const IWorkloadFactory& GetWorkloadFactory(const Layer& layer) const; - RefWorkloadFactory m_CpuRef; - NeonWorkloadFactory m_CpuAcc; - ClWorkloadFactory m_GpuAcc; + using BackendPtrMap = std::unordered_map<BackendId, IBackendInternalUniquePtr>; + using WorkloadFactoryMap = std::unordered_map<BackendId, IBackendInternal::IWorkloadFactoryPtr>; + + BackendPtrMap m_Backends; + WorkloadFactoryMap m_WorkloadFactories; std::unique_ptr<OptimizedNetwork> m_OptimizedNetwork; WorkloadQueue m_InputQueue; |