1 // 2 // Copyright © 2017 Arm Ltd and Contributors. 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 35