diff options
author | David Beck <david.beck@arm.com> | 2018-10-22 13:16:00 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
commit | 056be3cbe0306f4e1409c10952a8a73676a4acdd (patch) | |
tree | c7bcbeed66468b49dde7cf16f77256cd128b584c /src/armnn/DeviceSpec.hpp | |
parent | a8e06ed540a934f966679e1ef1cf7acf295211b3 (diff) | |
download | armnn-056be3cbe0306f4e1409c10952a8a73676a4acdd.tar.gz |
IVGCVSW-1991 : refactor m_SupportedComputes in DeviceSpecs
Change-Id: Ied3d54dc356f5e4f87aeb59f66423ac1f893dd01
Diffstat (limited to 'src/armnn/DeviceSpec.hpp')
-rw-r--r-- | src/armnn/DeviceSpec.hpp | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/src/armnn/DeviceSpec.hpp b/src/armnn/DeviceSpec.hpp index af0d8f578b..834ce0921d 100644 --- a/src/armnn/DeviceSpec.hpp +++ b/src/armnn/DeviceSpec.hpp @@ -14,15 +14,19 @@ namespace armnn class DeviceSpec : public IDeviceSpec { public: - DeviceSpec() {} + DeviceSpec(const BackendIdSet& supportedBackends) + : m_SupportedBackends{supportedBackends} {} + virtual ~DeviceSpec() {} - virtual std::vector<IBackendSharedPtr> GetBackends() const + virtual const BackendIdSet& GetSupportedBackends() const { - return std::vector<IBackendSharedPtr>(); + return m_SupportedBackends; } - std::set<BackendId> m_SupportedComputeDevices; +private: + DeviceSpec() = delete; + BackendIdSet m_SupportedBackends; }; } |