blob: 202510346e5a1b7dd6e2d6b5fcdef60981d8ad04 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "RangeTracker.hpp"
namespace armnn
{
void RangeTracker::SetRange(const armnn::IConnectableLayer *layer, unsigned int outputIdx, float min, float max)
{
auto& ranges = m_GuidToRangesMap[layer->GetGuid()];
if (ranges.size() < layer->GetNumOutputSlots())
{
ranges.resize(layer->GetNumOutputSlots());
}
ranges[outputIdx] = std::make_pair(min, max);
}
RangeTracker::MinMaxRange RangeTracker::GetRange(armnn::LayerGuid guid, unsigned int idx) const
{
auto search = m_GuidToRangesMap.find(guid);
if (search == m_GuidToRangesMap.end())
{
return DefaultRange();
}
return search->second.at(idx);
}
} //namespace armnn
|
