From 033626d1ee9256032309bbc6685c3a69a919cd64 Mon Sep 17 00:00:00 2001
From: Les Bell <les.bell@arm.com>
Date: Mon, 3 Sep 2018 16:24:52 +0100
Subject: IVGCVSW-1783 refactor add/mul/div layers

Change-Id: Ic043030455b3cf8ad7f92fd0a75175c1827a95f4
---
 src/armnn/layers/AdditionLayer.hpp | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

(limited to 'src/armnn/layers/AdditionLayer.hpp')

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;
-- 
cgit v1.2.1