1 files changed, 3 insertions, 7 deletions
diff --git a/src/armnn/layers/AdditionLayer.hpp b/src/armnn/layers/AdditionLayer.hpp
index 37f0b5c259..d34536d114 100644
--- a/src/armnn/layers/AdditionLayer.hpp
+++ b/src/armnn/layers/AdditionLayer.hpp
@@ -4,23 +4,19 @@
 //
 #pragma once
 
-#include <Layer.hpp>
+#include "ArithmeticBaseLayer.hpp"
 
 namespace armnn
 {
 
-class AdditionLayer : public Layer
+class AdditionLayer : public ArithmeticBaseLayer
 {
 public:
-    virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph&            graph,
+    virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph,
                                                       const IWorkloadFactory& factory) const override;
 
     AdditionLayer* Clone(Graph& graph) const override;
 
-    void ValidateTensorShapesFromInputs() override;
-
-    std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
-
 protected:
     AdditionLayer(const char* name);
     ~AdditionLayer() = default;