IVGCVSW-1783 refactor add/mul/div layers

Change-Id: Ic043030455b3cf8ad7f92fd0a75175c1827a95f4

1 files changed, 3 insertions, 6 deletions

diff --git a/src/armnn/layers/MultiplicationLayer.hpp b/src/armnn/layers/MultiplicationLayer.hpp
index bbfd1ee694..264d12feeb 100644
--- a/src/armnn/layers/MultiplicationLayer.hpp
+++ b/src/armnn/layers/MultiplicationLayer.hpp
@@ -4,22 +4,19 @@
 //
 #pragma once
 
-#include <Layer.hpp>
+#include "ArithmeticBaseLayer.hpp"
 
 namespace armnn
 {
 
-class MultiplicationLayer : public Layer
+class MultiplicationLayer : 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;
 
     MultiplicationLayer* Clone(Graph& graph) const override;
 
-    void ValidateTensorShapesFromInputs() override;
-    std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
-
 protected:
     MultiplicationLayer(const char* name);
     ~MultiplicationLayer() = default;