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 16 class QuantizeLayer : public Layer 17 { 18 public: 19 virtual std::unique_ptr<IWorkload> CreateWorkload(const IWorkloadFactory& factory) const override; 20 21 Layer* Clone(Graph& graph) const override; 22 23 void ValidateTensorShapesFromInputs() override; 24 25 void ExecuteStrategy(IStrategy& strategy) const override; 26 27 28 protected: 29 QuantizeLayer(const char* name); 30 ~QuantizeLayer() = default; 31 32 }; 33 34 } //namespace armnn 35