1 files changed, 15 insertions, 2 deletions
diff --git a/arm_compute/graph2/backends/NEON/NENodeValidator.h b/arm_compute/graph2/backends/NEON/NENodeValidator.h
index 8e8448526a..d39ab15a27 100644
--- a/arm_compute/graph2/backends/NEON/NENodeValidator.h
+++ b/arm_compute/graph2/backends/NEON/NENodeValidator.h
@@ -24,15 +24,28 @@
 #ifndef __ARM_COMPUTE_GRAPH2_NENODEVALIDATOR_H__
 #define __ARM_COMPUTE_GRAPH2_NENODEVALIDATOR_H__
 
-#include "arm_compute/graph2/INodeVisitor.h"
+#include "arm_compute/core/Error.h"
 
 namespace arm_compute
 {
 namespace graph2
 {
+// Forward declarations
+class INode;
+
 namespace backends
 {
-// TODO (geopin01) : Add node validator
+class NENodeValidator final
+{
+public:
+    /** Validate a node
+     *
+     * @param[in] node Node to validate
+     *
+     * @return An error status
+     */
+    static Status validate(INode *node);
+};
 } // namespace backends
 } // namespace graph2
 } // namespace arm_compute