From dcb751f338b86c811a35bd0a9413ba9b8df7b718 Mon Sep 17 00:00:00 2001
From: David Beck <david.beck@arm.com>
Date: Wed, 3 Oct 2018 11:42:42 +0100
Subject: IVGCVSW-1642 : introducing the IBackend interface

Change-Id: Iaadee0a08c0594c9a3c802a48fe346e15f2cbbb2
---
 src/backends/WorkloadFactory.hpp | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

(limited to 'src/backends/WorkloadFactory.hpp')

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;
-- 
cgit v1.2.1