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/OverrideInputRangeVisitor.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/OverrideInputRangeVisitor.hpp')
-rw-r--r-- | src/armnn/OverrideInputRangeVisitor.hpp | 93 |
1 files changed, 0 insertions, 93 deletions
diff --git a/src/armnn/OverrideInputRangeVisitor.hpp b/src/armnn/OverrideInputRangeVisitor.hpp deleted file mode 100644 index 196a3aab1d..0000000000 --- a/src/armnn/OverrideInputRangeVisitor.hpp +++ /dev/null @@ -1,93 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include "NetworkQuantizer.hpp" -#include "armnn/LayerVisitorBase.hpp" -#include "RangeTracker.hpp" - -#include <unordered_map> - -namespace armnn -{ -class OverrideInputRangeStrategy : public IStrategy -{ -private: - using MinMaxRange = RangeTracker::MinMaxRange; -public : - OverrideInputRangeStrategy(RangeTracker& ranges, - LayerBindingId layerId, - const MinMaxRange& minMaxRange) - : m_Ranges(ranges) - , m_LayerId(layerId) - , m_MinMaxRange(minMaxRange){} - - ~OverrideInputRangeStrategy() = default; - - void ExecuteStrategy(const armnn::IConnectableLayer* layer, - const BaseDescriptor& descriptor, - const std::vector<armnn::ConstTensor>& constants, - const char* name, - const armnn::LayerBindingId id) override - { - IgnoreUnused(name, constants, id, descriptor); - - switch (layer->GetType()) - { - case armnn::LayerType::Input : - { - if (m_LayerId == id) - { - m_Ranges.SetRange(layer, 0, m_MinMaxRange.first, m_MinMaxRange.second); - } - break; - } - default: - { - std::cout << "dont know this one" << std::endl; - } - } - } - -private: - /// Mapping from a layer Guid to an array of ranges for outputs - RangeTracker& m_Ranges; - - /// The id of the input layer of which to override the input range - LayerBindingId m_LayerId; - - /// The new input range to be applied to the input layer - MinMaxRange m_MinMaxRange; -}; - - - -/// Visitor object for overriding the input range of the quantized input layers in a network -class OverrideInputRangeVisitor : public LayerVisitorBase<VisitorNoThrowPolicy> -{ -private: - using MinMaxRange = RangeTracker::MinMaxRange; - -public: - OverrideInputRangeVisitor(RangeTracker& ranges, - LayerBindingId layerId, - const MinMaxRange& minMaxRange); - ~OverrideInputRangeVisitor() = default; - - void VisitInputLayer(const IConnectableLayer* layer, LayerBindingId id, const char* name = nullptr) override; - -private: - /// Mapping from a layer Guid to an array of ranges for outputs - RangeTracker& m_Ranges; - - /// The id of the input layer of which to override the input range - LayerBindingId m_LayerId; - - /// The new input range to be applied to the input layer - MinMaxRange m_MinMaxRange; -}; - -} // namespace armnn |