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