diff options
Diffstat (limited to 'src/armnn/layers/AdditionLayer.hpp')
| -rw-r--r-- | src/armnn/layers/AdditionLayer.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/layers/AdditionLayer.hpp b/src/armnn/layers/AdditionLayer.hpp index 211a2ef65f..9cdf09fb4e 100644 --- a/src/armnn/layers/AdditionLayer.hpp +++ b/src/armnn/layers/AdditionLayer.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 AdditionLayer : public ArithmeticBaseLayer +class AdditionLayer : public ElementwiseBaseLayer { public: virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph, |