diff options
Diffstat (limited to 'src/armnn/layers/MaximumLayer.cpp')
| -rw-r--r-- | src/armnn/layers/MaximumLayer.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/layers/MaximumLayer.cpp b/src/armnn/layers/MaximumLayer.cpp index 67a234216c..c0da8d6ef8 100644 --- a/src/armnn/layers/MaximumLayer.cpp +++ b/src/armnn/layers/MaximumLayer.cpp @@ -14,8 +14,9 @@ namespace armnn { MaximumLayer::MaximumLayer(const char* name) -: ArithmeticBaseLayer(2, 1, LayerType::Maximum, name) -{} + : ElementwiseBaseLayer(2, 1, LayerType::Maximum, name) +{ +} std::unique_ptr<IWorkload> MaximumLayer::CreateWorkload(const Graph& graph, const IWorkloadFactory& factory) const |