TensorPrinter Struct Reference

#include <NetworkExecutionUtils.hpp>

Public Member Functions
	TensorPrinter (const std::string &binding, const armnn::TensorInfo &info, const std::string &outputTensorFile, bool dequantizeOutput)
void	operator() (const std::vector< float > &values)
void	operator() (const std::vector< uint8_t > &values)
void	operator() (const std::vector< int > &values)

Detailed Description

Definition at line 24 of file NetworkExecutionUtils.hpp.

Constructor & Destructor Documentation

TensorPrinter()

TensorPrinter	(	const std::string &	binding,
		const armnn::TensorInfo &	info,
		const std::string &	outputTensorFile,
		bool	dequantizeOutput
	)

Definition at line 92 of file NetworkExecutionUtils.cpp.

                             : m_OutputBinding(binding)
                             , m_Scale(info.GetQuantizationScale())
                             , m_Offset(info.GetQuantizationOffset())
                             , m_OutputTensorFile(outputTensorFile)
                             , m_DequantizeOutput(dequantizeOutput) {}

Member Function Documentation

operator()() [1/3]

void operator()

(

const std::vector< float > &

values

)

Definition at line 102 of file NetworkExecutionUtils.cpp.

Referenced by operator()().

{
    ForEachValue(values, [](float value)
    {
        printf("%f ", value);
    });
    WriteToFile(values);
}

operator()() [2/3]

void operator()

(

const std::vector< uint8_t > &

values

)

Definition at line 111 of file NetworkExecutionUtils.cpp.

References armnn::Dequantize(), and operator()().

{
    if(m_DequantizeOutput)
    {
        auto& scale = m_Scale;
        auto& offset = m_Offset;
        std::vector<float> dequantizedValues;
        ForEachValue(values, [&scale, &offset, &dequantizedValues](uint8_t value)
        {
            auto dequantizedValue = armnn::Dequantize(value, scale, offset);
            printf("%f ", dequantizedValue);
            dequantizedValues.push_back(dequantizedValue);
        });
        WriteToFile(dequantizedValues);
    }
    else
    {
        const std::vector<int> intValues(values.begin(), values.end());
        operator()(intValues);
    }
}

operator()() [3/3]

void operator()

(

const std::vector< int > &

values

)

Definition at line 133 of file NetworkExecutionUtils.cpp.

References ARMNN_LOG.

{
    ForEachValue(values, [](int value)
    {
        printf("%d ", value);
    });
    WriteToFile(values);
}

---

The documentation for this struct was generated from the following files:

• tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp
• tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp