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