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/DivisionLayer.hpp | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

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

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