diff options
| author | Les Bell <les.bell@arm.com> | 2018-09-03 16:24:52 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-17 17:21:25 +0100 |
| commit | 033626d1ee9256032309bbc6685c3a69a919cd64 (patch) | |
| tree | 42a1ab9c3b83c4da019b69e2189846663e0d19be /src/armnn/layers/DivisionLayer.hpp | |
| parent | 9e53a35b66b1ec7ceee7c712380a13596175b83b (diff) | |
| download | armnn-033626d1ee9256032309bbc6685c3a69a919cd64.tar.gz | |
IVGCVSW-1783 refactor add/mul/div layers
Change-Id: Ic043030455b3cf8ad7f92fd0a75175c1827a95f4
Diffstat (limited to 'src/armnn/layers/DivisionLayer.hpp')
| -rw-r--r-- | src/armnn/layers/DivisionLayer.hpp | 7 |
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; |