| // |
| // 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 |
| { |
| |
| /// 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 |