ArmNN  NotReleased
QuantizeLayer.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <Layer.hpp>
9 
10 namespace armnn {
11 
12 //Forward
13 class IWorkload;
14 class IWorkloadFactory;
15 class ILayerVisitor;
16 
17 class QuantizeLayer : public Layer
18 {
19 public:
20  virtual std::unique_ptr<IWorkload> CreateWorkload(const IWorkloadFactory& factory) const override;
21 
22  Layer* Clone(Graph& graph) const override;
23 
24  void ValidateTensorShapesFromInputs() override;
25 
26  void Accept(ILayerVisitor& visitor) const override;
27 
28 protected:
29  QuantizeLayer(const char* name);
30  ~QuantizeLayer() = default;
31 
32 };
33 
34 } //namespace armnn
~QuantizeLayer()=default
QuantizeLayer(const char *name)
Layer * Clone(Graph &graph) const override
void Accept(ILayerVisitor &visitor) const override
void ValidateTensorShapesFromInputs() override
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override