diff options
author | Kevin May <kevin.may@arm.com> | 2021-04-22 16:51:18 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2021-04-27 15:02:45 +0000 |
commit | 4a621c43174b6bdd9dc0bff839b245bc2139d6a6 (patch) | |
tree | 88b2bbc803284e7c5051c8812a468e5f3c834b9c /src/armnn/NetworkQuantizer.hpp | |
parent | 65542e4f250a66fd0aba0c1b509e11cb518ee2d5 (diff) | |
download | armnn-4a621c43174b6bdd9dc0bff839b245bc2139d6a6.tar.gz |
IVGCVSW-5719 Remove Quantizer
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I8a29df03afdd6f1cc8413b219e179272cd2d51cf
Diffstat (limited to 'src/armnn/NetworkQuantizer.hpp')
-rw-r--r-- | src/armnn/NetworkQuantizer.hpp | 61 |
1 files changed, 0 insertions, 61 deletions
diff --git a/src/armnn/NetworkQuantizer.hpp b/src/armnn/NetworkQuantizer.hpp deleted file mode 100644 index a07ac8827e..0000000000 --- a/src/armnn/NetworkQuantizer.hpp +++ /dev/null @@ -1,61 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include <armnn/INetwork.hpp> -#include <armnnQuantizer/INetworkQuantizer.hpp> -#include <armnn/IRuntime.hpp> -#include <armnn/Types.hpp> -#include <armnn/Optional.hpp> - -#include "DynamicQuantizationStrategy.hpp" -#include "RangeTracker.hpp" - -namespace armnn -{ - -class NetworkQuantizer : public INetworkQuantizer -{ -public: - NetworkQuantizer(INetwork* inputNetwork, const QuantizerOptions& options) - : m_InputNetwork(inputNetwork), - m_NetworkId(0), - m_Runtime(nullptr, &IRuntime::Destroy), - m_RefineCount(0), - m_Options(options) {} - - void OverrideInputRange(LayerBindingId layerId, float min, float max) override; - void Refine(const InputTensors& inputTensors) override; - - // Required for testing? Need some way to get min/max in RangeTracker (m_Ranges) - std::pair<float, float> GetMinMaxRange(LayerGuid guid, unsigned int idx) { return m_Ranges.GetRange(guid, idx); } - INetworkPtr ExportNetwork() override; - -private: - /// Original input network to quantize - INetwork* m_InputNetwork; - - NetworkId m_NetworkId; - - // if we are run in dynamic mode this unique pointer will hold - // the runtime between invocations of the Refine method. - IRuntimePtr m_Runtime; - - Optional<DynamicQuantizationStrategy> m_DynamicQuantizationStrategy; - - // counts the number of times refine is called - unsigned int m_RefineCount; - - /// Mapping from Guid to an array of ranges for outputs - RangeTracker m_Ranges; - - /// Options for the NetworkQuantizer - QuantizerOptions m_Options; - - std::pair<float, float> FindMinMax(ITensorHandle* tensorHandle); -}; - -} //namespace armnn |