RangeTracker.hpp

Go to the documentation of this file.

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#pragma once

#include <armnn/INetwork.hpp>
#include <armnn/Types.hpp>

#include <utility>
#include <unordered_map>

namespace armnn
{

class RangeTracker
{
public:
    using MinMaxRange  = std::pair<float, float>;

    MinMaxRange GetRange(LayerGuid guid, unsigned int idx) const;

    void SetRange(const IConnectableLayer* layer, unsigned int outputIdx, float min, float max);

    bool IsEmpty() const { return m_GuidToRangesMap.empty(); }

    bool HasRanges(LayerGuid guid) const { return m_GuidToRangesMap.find(guid) != m_GuidToRangesMap.end(); }

    void RefineMin(LayerGuid guid, unsigned int slotIndex, float newMin);

    void RefineMax(LayerGuid guid, unsigned int slotIndex, float newMax);

    void ResetMinMax(LayerGuid guid, unsigned int idx, float newMin, float newMax);

    void Reset();

    void SetDynamicMode(bool flag) { m_DynamicMode = flag; }

    bool IsInDynamicMode() const { return m_DynamicMode; }

private:
    using MinMaxRanges = std::vector<MinMaxRange>;

    MinMaxRange DefaultRange() const { return std::make_pair(-15.0f, 15.0f); }

    std::unordered_map<LayerGuid, MinMaxRanges> m_GuidToRangesMap;

    bool m_DynamicMode = false;
};

} //namespace armnn