1 files changed, 13 insertions, 5 deletions
diff --git a/src/backends/neon/workloads/NeonSplitterWorkload.hpp b/src/backends/neon/workloads/NeonSplitterWorkload.hpp
index 2a7ee193d0..f9025663ca 100644
--- a/src/backends/neon/workloads/NeonSplitterWorkload.hpp
+++ b/src/backends/neon/workloads/NeonSplitterWorkload.hpp
@@ -7,18 +7,26 @@
 
 #include <backendsCommon/Workload.hpp>
 
+#include <arm_compute/runtime/NEON/NEFunctions.h>
+
+#include <functional>
+
 namespace armnn
 {
 
+arm_compute::Status NeonSplitterWorkloadValidate(const TensorInfo& input,
+                                                 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
+                                                 unsigned int splitAxis);
+
 class NeonSplitterWorkload : public BaseWorkload<SplitterQueueDescriptor>
 {
 public:
-    using BaseWorkload<SplitterQueueDescriptor>::BaseWorkload;
+    NeonSplitterWorkload(const SplitterQueueDescriptor& descriptor, const WorkloadInfo& info);
+
+    void Execute() const override;
 
-    virtual void Execute() const override
-    {
-        // With subtensors, splitter is a no-op.
-    }
+private:
+    mutable std::unique_ptr<arm_compute::NESplit> m_Layer;
 };
 
 } //namespace armnn