18 std::vector<unsigned int>
ParseArray(std::istream& stream);
21 std::vector<std::string>
ParseStringList(
const std::string& inputString,
const char* delimiter);
27 const std::string& outputTensorFile,
28 bool dequantizeOutput,
29 bool printToConsole =
true);
31 void operator()(
const std::vector<float>& values);
33 void operator()(
const std::vector<uint8_t>& values);
35 void operator()(
const std::vector<int>& values);
37 void operator()(
const std::vector<int8_t>& values);
40 template<
typename Container,
typename Delegate>
41 void ForEachValue(
const Container& c, Delegate delegate);
44 void WriteToFile(
const std::vector<T>& values);
46 std::string m_OutputBinding;
49 std::string m_OutputTensorFile;
50 bool m_DequantizeOutput;
51 bool m_PrintToConsole;
57 unsigned int numElements,
58 const std::string& dataTypeStr,
68 bool ValidatePath(
const std::string& file,
const bool expectFile);
76 bool ValidatePaths(
const std::vector<std::string>& fileVec,
const bool expectFile);
78 template<
typename T,
typename TParseElementFunc>
79 std::vector<T>
ParseArrayImpl(std::istream& stream, TParseElementFunc parseElementFunc,
const char* chars =
"\t ,:")
81 std::vector<T> result;
84 while (std::getline(stream, line))
87 for (
const std::string& token : tokens)
93 result.push_back(parseElementFunc(token));
95 catch (
const std::exception&)
97 ARMNN_LOG(error) <<
"'" << token <<
"' is not a valid number. It has been ignored.";
106 template <
typename T,
typename TParseElementFunc>
108 unsigned int numElements,
110 TParseElementFunc parseFunction)
112 const bool readFromFile = dataFile.
has_value() && !dataFile.
value().empty();
114 std::ifstream inputTensorFile;
117 inputTensorFile = std::ifstream(dataFile.
value());
120 tensorData = readFromFile ?
121 ParseArrayImpl<T>(inputTensorFile, parseFunction) :
122 std::vector<T>(numElements, static_cast<T>(0));
std::vector< std::string > StringTokenizer(const std::string &str, const char *delimiters, bool tokenCompression=true)
Function to take a string and a list of delimiters and split the string into tokens based on those de...
bool ValidatePath(const std::string &file, const bool expectFile)
Verifies if the given string is a valid path.
void operator()(const std::vector< float > &values)
#define ARMNN_LOG(severity)
std::vector< unsigned int > ParseArray(std::istream &stream)
std::vector< std::string > ParseStringList(const std::string &inputString, const char *delimiter)
Splits a given string at every accurance of delimiter into a vector of string.
bool has_value() const noexcept
std::pair< float, int32_t > QuantizationParams
void PopulateTensorWithData(armnnUtils::TContainer &tensorData, unsigned int numElements, const std::string &dataTypeStr, const armnn::Optional< QuantizationParams > &qParams, const armnn::Optional< std::string > &dataFile)
TensorPrinter(const std::string &binding, const armnn::TensorInfo &info, const std::string &outputTensorFile, bool dequantizeOutput, bool printToConsole=true)
std::vector< T > ParseArrayImpl(std::istream &stream, TParseElementFunc parseElementFunc, const char *chars="\,:")
bool ValidatePaths(const std::vector< std::string > &fileVec, const bool expectFile)
Verifies if a given vector of strings are valid paths.
mapbox::util::variant< std::vector< float >, std::vector< int >, std::vector< unsigned char >, std::vector< int8_t > > TContainer
void PopulateTensorWithDataGeneric(std::vector< T > &tensorData, unsigned int numElements, const armnn::Optional< std::string > &dataFile, TParseElementFunc parseFunction)