diff options
author | David Beck <david.beck@arm.com> | 2018-10-03 11:42:42 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:58 +0100 |
commit | dcb751f338b86c811a35bd0a9413ba9b8df7b718 (patch) | |
tree | bda3027fe7d958383eaa85105f611bfc07c3838c /src/backends/WorkloadFactory.hpp | |
parent | 3041e3036398048b4f8b72e0a321d77cd222a2fa (diff) | |
download | armnn-dcb751f338b86c811a35bd0a9413ba9b8df7b718.tar.gz |
IVGCVSW-1642 : introducing the IBackend interface
Change-Id: Iaadee0a08c0594c9a3c802a48fe346e15f2cbbb2
Diffstat (limited to 'src/backends/WorkloadFactory.hpp')
-rw-r--r-- | src/backends/WorkloadFactory.hpp | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/src/backends/WorkloadFactory.hpp b/src/backends/WorkloadFactory.hpp index 77e810c9ad..38448ca378 100644 --- a/src/backends/WorkloadFactory.hpp +++ b/src/backends/WorkloadFactory.hpp @@ -4,10 +4,10 @@ // #pragma once -#include "Workload.hpp" #include <memory> -#include "armnn/TensorFwd.hpp" -#include "OutputHandler.hpp" +#include <armnn/TensorFwd.hpp> +#include <backends/OutputHandler.hpp> +#include <backends/Workload.hpp> #include <boost/optional.hpp> namespace armnn @@ -32,9 +32,13 @@ public: /// Inform the memory manager to acquire memory virtual void Acquire() { } - static bool IsLayerSupported(Compute compute, const Layer& layer, boost::optional<DataType> dataType, + static bool IsLayerSupported(Compute compute, + const IConnectableLayer& layer, + boost::optional<DataType> dataType, std::string& outReasonIfUnsupported); - static bool IsLayerSupported(const Layer& layer, boost::optional<DataType> dataType, + + static bool IsLayerSupported(const IConnectableLayer& layer, + boost::optional<DataType> dataType, std::string& outReasonIfUnsupported); virtual bool SupportsSubTensors() const = 0; |