blob: bb7aff1c2ce2d48a1809ce1b9c688b4a0e59fc9c (
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "QuantizationInput.hpp"
#include <iostream>
#include <fstream>
#include <cstring>
#include "armnn/Exceptions.hpp"
namespace armnnQuantizer
{
QuantizationInput::QuantizationInput(const unsigned int passId,
const armnn::LayerBindingId bindingId,
const std::string fileName):
m_PassId(passId)
{
m_LayerBindingIdToFileName.emplace(bindingId, fileName);
}
QuantizationInput::QuantizationInput(const QuantizationInput& other)
{
m_PassId = other.GetPassId();
m_LayerBindingIdToFileName.clear();
for (armnn::LayerBindingId bindingId : other.GetLayerBindingIds())
{
std::string filename = other.GetFileName(bindingId);
AddEntry(bindingId, filename);
}
}
void QuantizationInput::AddEntry(const armnn::LayerBindingId bindingId, const std::string fileName)
{
m_LayerBindingIdToFileName.emplace(bindingId, fileName);
}
std::vector<float> QuantizationInput::GetDataForEntry(const armnn::LayerBindingId bindingId) const
{
if (m_LayerBindingIdToFileName.at(bindingId).empty())
{
throw armnn::Exception("Layer binding ID not found");
}
std::string fileName = m_LayerBindingIdToFileName.at(bindingId);
std::ifstream in(fileName.c_str(), std::ifstream::binary);
if (!in.is_open())
{
throw armnn::Exception("Failed to open input tensor file " + fileName);
}
std::string line;
std::vector<float> values;
char* pEnd;
while (std::getline(in, line, ' '))
{
values.emplace_back(std::strtof(line.c_str(), &pEnd));
}
return values;
}
std::vector<armnn::LayerBindingId> QuantizationInput::GetLayerBindingIds() const
{
std::vector<armnn::LayerBindingId> layerBindingIDs;
for (auto iterator = m_LayerBindingIdToFileName.begin(); iterator != m_LayerBindingIdToFileName.end(); ++iterator)
{
layerBindingIDs.emplace_back(iterator->first);
}
return layerBindingIDs;
}
unsigned long QuantizationInput::GetNumberOfInputs() const
{
return m_LayerBindingIdToFileName.size();
}
unsigned int QuantizationInput::GetPassId() const
{
return m_PassId;
}
std::string QuantizationInput::GetFileName(const armnn::LayerBindingId bindingId) const
{
auto iterator = m_LayerBindingIdToFileName.find(bindingId);
if (iterator != m_LayerBindingIdToFileName.end())
{
return m_LayerBindingIdToFileName.at(bindingId);
}
else
{
throw armnn::Exception("Could not retrieve filename for binding ID " + std::to_string(bindingId));
}
}
QuantizationInput::~QuantizationInput() noexcept
{
}
}
|
