1 // 2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include "NeonBaseWorkload.hpp" 9 10 #include <arm_compute/runtime/NEON/functions/NEQuantizationLayer.h> 11 12 namespace armnn { 13 14 arm_compute::Status NeonQuantizeWorkloadValidate(const TensorInfo& input, const TensorInfo& output); 15 16 class NeonQuantizeWorkload : public NeonBaseWorkload<QuantizeQueueDescriptor> 17 { 18 public: 19 NeonQuantizeWorkload(const QuantizeQueueDescriptor& descriptor, const WorkloadInfo& workloadInfo); 20 void Execute() const override; 21 22 private: 23 mutable std::unique_ptr<arm_compute::NEQuantizationLayer> m_Layer; 24 }; 25 26 } // namespace armnn 27