QuantizationInput.hpp

Go to the documentation of this file.

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

#pragma once

#include <map>
#include <armnn/Types.hpp>
#include <armnnQuantizer/INetworkQuantizer.hpp>

namespace armnnQuantizer
{

class QuantizationInput
{
public:

    QuantizationInput(const unsigned int passId,
                      const armnn::LayerBindingId bindingId,
                      const std::string fileName);

    QuantizationInput(const QuantizationInput& other);

    // Add binding ID to image tensor filepath entry
    void AddEntry(const armnn::LayerBindingId bindingId, const std::string fileName);

    // Retrieve tensor data for entry with provided binding ID
    std::vector<float> GetDataForEntry(const armnn::LayerBindingId bindingId) const;

    std::vector<armnn::LayerBindingId> GetLayerBindingIds() const;

    unsigned long GetNumberOfInputs() const;

    unsigned int GetPassId() const;

    std::string GetFileName(const armnn::LayerBindingId bindingId) const;

    ~QuantizationInput() noexcept;

private:
    unsigned int m_PassId;
    std::map<armnn::LayerBindingId, std::string> m_LayerBindingIdToFileName;

};

}