diff options
author | Nina Drozd <nina.drozd@arm.com> | 2019-04-25 15:45:20 +0100 |
---|---|---|
committer | Nina Drozd <nina.drozd@arm.com> | 2019-05-03 14:43:50 +0100 |
commit | 59e15b00ea51fee4baeea750dc11ab1952dfab1d (patch) | |
tree | 97e6c9230bf153d404ad3c3e0e285acdb0b6232d /src/armnn | |
parent | 8b194fbe79d44cba566ad8b508d1c8902987ae3c (diff) | |
download | armnn-59e15b00ea51fee4baeea750dc11ab1952dfab1d.tar.gz |
IVGCVSW-2834 Add dynamic quantization via datasets
* Add QuantizationDataSet class for quantization data parsed from CSV file
* Add QuantizationInput for retrieving quantization data for each layer ID
* Add unit tests for command line processor and QuantizationDataSet
Change-Id: Iaf0a747b5f25a59a766ac04f7158e8cb7909d179
Signed-off-by: Nina Drozd <nina.drozd@arm.com>
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/NetworkQuantizer.cpp | 5 | ||||
-rw-r--r-- | src/armnn/NetworkQuantizer.hpp | 1 |
2 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/NetworkQuantizer.cpp b/src/armnn/NetworkQuantizer.cpp index f577aea00e..4692a6803f 100644 --- a/src/armnn/NetworkQuantizer.cpp +++ b/src/armnn/NetworkQuantizer.cpp @@ -49,6 +49,11 @@ void NetworkQuantizer::OverrideInputRange(LayerBindingId layerId, float min, flo VisitLayers(inputLayers, overrideInputRangeVisitor); } +void NetworkQuantizer::Refine(const InputTensors& inputTensors) +{ + //Implementation in a following commit +} + INetworkPtr NetworkQuantizer::ExportNetwork() { const Graph& graph = boost::polymorphic_downcast<const Network*>(m_InputNetwork)->GetGraph().TopologicalSort(); diff --git a/src/armnn/NetworkQuantizer.hpp b/src/armnn/NetworkQuantizer.hpp index 5e93f70290..4f6359f36d 100644 --- a/src/armnn/NetworkQuantizer.hpp +++ b/src/armnn/NetworkQuantizer.hpp @@ -21,6 +21,7 @@ public: : m_InputNetwork(inputNetwork), m_Options(options) {} void OverrideInputRange(LayerBindingId layerId, float min, float max) override; + void Refine(const InputTensors& inputTensors) override; INetworkPtr ExportNetwork() override; private: |