aboutsummaryrefslogtreecommitdiff
path: root/include
diff options
context:
space:
mode:
Diffstat (limited to 'include')
-rw-r--r--include/armnn/BackendId.hpp2
-rw-r--r--include/armnn/ILayerSupport.hpp3
-rw-r--r--include/armnn/Types.hpp3
3 files changed, 6 insertions, 2 deletions
diff --git a/include/armnn/BackendId.hpp b/include/armnn/BackendId.hpp
index d2479ebb2c..af3b7995eb 100644
--- a/include/armnn/BackendId.hpp
+++ b/include/armnn/BackendId.hpp
@@ -4,10 +4,12 @@
//
#pragma once
+#include <ostream>
#include <set>
#include <unordered_set>
#include <string>
#include <memory>
+#include <vector>
namespace armnn
{
diff --git a/include/armnn/ILayerSupport.hpp b/include/armnn/ILayerSupport.hpp
index 58adea85de..d63c3a7063 100644
--- a/include/armnn/ILayerSupport.hpp
+++ b/include/armnn/ILayerSupport.hpp
@@ -8,6 +8,7 @@
#include <armnn/Optional.hpp>
#include <vector>
#include <cctype>
+#include <memory>
namespace armnn
{
@@ -178,4 +179,6 @@ public:
Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
}; // class ILayerSupport
+using ILayerSupportSharedPtr = std::shared_ptr<ILayerSupport>;
+
} // namespace armnn
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp
index bb0b1e6ca7..cd6e17be37 100644
--- a/include/armnn/Types.hpp
+++ b/include/armnn/Types.hpp
@@ -6,8 +6,8 @@
#include <array>
#include <memory>
-#include "ILayerSupport.hpp"
#include "BackendId.hpp"
+#include "Exceptions.hpp"
namespace armnn
{
@@ -148,7 +148,6 @@ protected:
public:
virtual const BackendId& GetId() const = 0;
- virtual const ILayerSupport& GetLayerSupport() const = 0;
};
using IBackendSharedPtr = std::shared_ptr<IBackend>;