IVGCVSW-2771 Fix SubTensor error in vgg16 ExecuteNetwork CL

 * Add check if Sub-tensors cannot be used, call ACL function
 * Add ClSplitterWorkload functions
 * Modify IsSplitterSupported to call ACL validate function
if sub-tensor cannot be used
 * Also check if quantization parameters match when using sub-tensors

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I5dfd0e422b7d485dd4421a664add83d870bec5d6

1 files changed, 13 insertions, 6 deletions

diff --git a/src/backends/cl/workloads/ClSplitterWorkload.hpp b/src/backends/cl/workloads/ClSplitterWorkload.hpp
index 950335902e..d024452d78 100644
--- a/src/backends/cl/workloads/ClSplitterWorkload.hpp
+++ b/src/backends/cl/workloads/ClSplitterWorkload.hpp
@@ -7,19 +7,26 @@
 
 #include <backendsCommon/Workload.hpp>
 
+#include <arm_compute/runtime/CL/CLFunctions.h>
+
+#include <functional>
+
 namespace armnn
 {
 
-// Base class template providing an implementation of the Splitter layer common to all data types.
+arm_compute::Status ClSplitterWorkloadValidate(const TensorInfo& input,
+                                               const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
+                                               unsigned int splitAxis);
+
 class ClSplitterWorkload : public BaseWorkload<SplitterQueueDescriptor>
 {
 public:
-    using BaseWorkload<SplitterQueueDescriptor>::BaseWorkload;
+    ClSplitterWorkload(const SplitterQueueDescriptor& descriptor, const WorkloadInfo& info);
+
+    void Execute() const override;
 
-    void Execute() const override
-    {
-        // With subtensors, splitter is a no-op.
-    }
+private:
+    mutable std::unique_ptr<arm_compute::CLSplit> m_Layer;
 };
 
 } //namespace armnn