1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "OverrideInputRangeVisitor.hpp"
7 #include "NetworkQuantizerUtils.hpp"
8 #include "Layer.hpp"
9
10 #include <armnn/utility/IgnoreUnused.hpp>
11
12 namespace armnn
13 {
14
OverrideInputRangeVisitor(RangeTracker & ranges,LayerBindingId layerId,const MinMaxRange & minMaxRange)15 OverrideInputRangeVisitor::OverrideInputRangeVisitor(RangeTracker& ranges,
16 LayerBindingId layerId,
17 const MinMaxRange& minMaxRange)
18 : m_Ranges(ranges)
19 , m_LayerId(layerId)
20 , m_MinMaxRange(minMaxRange)
21 {}
22
VisitInputLayer(const IConnectableLayer * layer,LayerBindingId id,const char * name)23 void OverrideInputRangeVisitor::VisitInputLayer(const IConnectableLayer* layer, LayerBindingId id, const char* name)
24 {
25 IgnoreUnused(name);
26 if (m_LayerId == id)
27 {
28 m_Ranges.SetRange(layer, 0, m_MinMaxRange.first, m_MinMaxRange.second);
29 }
30 }
31
32 } // namespace armnn
33