src/armnn/layers/MaximumLayer.hpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5 #pragma once
   6
   7 #include "ArithmeticBaseLayer.hpp"
   8
   9 namespace armnn
  10 {
  11
  12 class MaximumLayer : public ArithmeticBaseLayer
  13 {
  14 public:
  15     virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph,
  16                                                       const IWorkloadFactory& factory) const override;
  17
  18     MaximumLayer* Clone(Graph& graph) const override;
  19
  20 protected:
  21     MaximumLayer(const char* name);
  22
  23     ~MaximumLayer() = default;
  24 };
  25
  26 } // namespace