1 files changed, 2 insertions, 5 deletions
diff --git a/src/armnn/layers/DivisionLayer.hpp b/src/armnn/layers/DivisionLayer.hpp
index 1bd69c4446..c6a642e318 100644
--- a/src/armnn/layers/DivisionLayer.hpp
+++ b/src/armnn/layers/DivisionLayer.hpp
@@ -4,12 +4,12 @@
 //
 #pragma once
 
-#include <Layer.hpp>
+#include "ArithmeticBaseLayer.hpp"
 
 namespace armnn
 {
 
-class DivisionLayer : public Layer
+class DivisionLayer : public ArithmeticBaseLayer
 {
 public:
     virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph,
@@ -17,9 +17,6 @@ public:
 
     DivisionLayer* Clone(Graph& graph) const override;
 
-    void ValidateTensorShapesFromInputs() override;
-    std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
-
 protected:
     DivisionLayer(const char* name);
     ~DivisionLayer() = default;