diff options
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; }; } |