blob: 47b893a7f7497d29c9c68f934d0672439646276a (
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
#include <map>
#include "QuantizationInput.hpp"
#include "armnn/LayerVisitorBase.hpp"
#include "armnn/Tensor.hpp"
namespace armnnQuantizer
{
/// QuantizationDataSet is a structure which is created after parsing a quantization CSV file.
/// It contains records of filenames which contain refinement data per pass ID for binding ID.
class QuantizationDataSet
{
using QuantizationInputs = std::vector<armnnQuantizer::QuantizationInput>;
public:
using iterator = QuantizationInputs::iterator;
using const_iterator = QuantizationInputs::const_iterator;
QuantizationDataSet();
QuantizationDataSet(std::string csvFilePath);
~QuantizationDataSet();
bool IsEmpty() const {return m_QuantizationInputs.empty();}
iterator begin() { return m_QuantizationInputs.begin(); }
iterator end() { return m_QuantizationInputs.end(); }
const_iterator begin() const { return m_QuantizationInputs.begin(); }
const_iterator end() const { return m_QuantizationInputs.end(); }
const_iterator cbegin() const { return m_QuantizationInputs.cbegin(); }
const_iterator cend() const { return m_QuantizationInputs.cend(); }
private:
void ParseCsvFile();
QuantizationInputs m_QuantizationInputs;
std::string m_CsvFilePath;
};
/// Visitor class implementation to gather the TensorInfo for LayerBindingID for creation of ConstTensor for Refine.
class InputLayerStrategy : public armnn::IStrategy
{
public:
virtual void ExecuteStrategy(const armnn::IConnectableLayer* layer,
const armnn::BaseDescriptor& descriptor,
const std::vector<armnn::ConstTensor>& constants,
const char* name,
const armnn::LayerBindingId id = 0) override;
armnn::TensorInfo GetTensorInfo(armnn::LayerBindingId);
private:
std::map<armnn::LayerBindingId, armnn::TensorInfo> m_TensorInfos;
};
/// Visitor class implementation to gather the TensorInfo for LayerBindingID for creation of ConstTensor for Refine.
class InputLayerVisitor : public armnn::LayerVisitorBase<armnn::VisitorNoThrowPolicy>
{
public:
void VisitInputLayer(const armnn::IConnectableLayer *layer, armnn::LayerBindingId id, const char* name);
armnn::TensorInfo GetTensorInfo(armnn::LayerBindingId);
private:
std::map<armnn::LayerBindingId, armnn::TensorInfo> m_TensorInfos;
};
} // namespace armnnQuantizer
|
