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