From a8d572dc48f47e66cd7abd6ad9b2d3a0f40ea94b Mon Sep 17 00:00:00 2001 From: Matteo Martincigh Date: Thu, 7 Feb 2019 17:51:09 +0000 Subject: IVGCVSW-2607 Implement Input range override mechanism * Added the OverrideInputRange method to the Quantizer API * Created OverrideInputRangeVisitor to implement the override mechanism * Moved the quantizer utility functions to the new NetworkQuantizerUtils files * Moved the map of quantization ranges out of the StaticRangeVisitor and into the NetworkQuantizer * Added unit tests * Code refactoring and cleanup Change-Id: I9c1d006c1b6a35fbc04584a832fbe489f8f9276d Signed-off-by: Matteo Martincigh --- include/armnn/INetworkQuantizer.hpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'include') diff --git a/include/armnn/INetworkQuantizer.hpp b/include/armnn/INetworkQuantizer.hpp index f3cc13ca35..5969fa4edf 100644 --- a/include/armnn/INetworkQuantizer.hpp +++ b/include/armnn/INetworkQuantizer.hpp @@ -7,9 +7,6 @@ #include -#include - - namespace armnn { @@ -23,11 +20,14 @@ public: static INetworkQuantizerPtr Create(INetwork* inputNetwork); ///< Create Quantizer object wrapped in unique_ptr static void Destroy(INetworkQuantizer* quantizer); ///< Destroy Quantizer object + /// Overrides the default quantization values for the input layer with the given id + virtual void OverrideInputRange(LayerBindingId layerId, float min, float max) = 0; + /// Extract final quantized network virtual INetworkPtr ExportNetwork() = 0; protected: - virtual ~INetworkQuantizer() {}; + virtual ~INetworkQuantizer() {} }; } //namespace armnn -- cgit v1.2.1