OverrideInputRangeVisitor.hpp

Go to the documentation of this file.

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

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:
    RangeTracker& m_Ranges;

    LayerBindingId m_LayerId;

    MinMaxRange m_MinMaxRange;
};

} // namespace armnn