From 4a621c43174b6bdd9dc0bff839b245bc2139d6a6 Mon Sep 17 00:00:00 2001 From: Kevin May Date: Thu, 22 Apr 2021 16:51:18 +0100 Subject: IVGCVSW-5719 Remove Quantizer Signed-off-by: Kevin May Change-Id: I8a29df03afdd6f1cc8413b219e179272cd2d51cf --- src/armnn/OverrideInputRangeVisitor.hpp | 93 --------------------------------- 1 file changed, 93 deletions(-) delete mode 100644 src/armnn/OverrideInputRangeVisitor.hpp (limited to 'src/armnn/OverrideInputRangeVisitor.hpp') 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 - -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& 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 -{ -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 -- cgit v1.2.1