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