diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/BackendId.hpp | 2 | ||||
-rw-r--r-- | include/armnn/ILayerSupport.hpp | 3 | ||||
-rw-r--r-- | include/armnn/Types.hpp | 3 |
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>; |