diff options
Diffstat (limited to 'src/armnn/layers/DivisionLayer.hpp')
| -rw-r--r-- | src/armnn/layers/DivisionLayer.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/layers/DivisionLayer.hpp b/src/armnn/layers/DivisionLayer.hpp index eaede4563b..158f8e8b5d 100644 --- a/src/armnn/layers/DivisionLayer.hpp +++ b/src/armnn/layers/DivisionLayer.hpp @@ -2,14 +2,15 @@ // Copyright © 2017 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // + #pragma once -#include "ArithmeticBaseLayer.hpp" +#include "ElementwiseBaseLayer.hpp" namespace armnn { -class DivisionLayer : public ArithmeticBaseLayer +class DivisionLayer : public ElementwiseBaseLayer { public: virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph, |