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