IVGCVSW-2601 Fix for ExecuteNetwork output data printing bug
Change-Id: I7e8d377cdbddf0176ae05d8b6b2fd03c6f614168
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp
index e6e6cdf..2978010 100644
--- a/tests/ExecuteNetwork/ExecuteNetwork.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp
@@ -140,9 +140,11 @@
void PrintOutputData(const std::string& outputLayerName, const std::vector<float>& data)
{
std::cout << outputLayerName << ": ";
- std::copy(data.begin(), data.end(),
- std::ostream_iterator<float>(std::cout, " "));
- std::cout << std::endl;
+ for (size_t i = 0; i < data.size(); i++)
+ {
+ printf("%f ", data[i]);
+ }
+ printf("\n");
}
void RemoveDuplicateDevices(std::vector<armnn::BackendId>& computeDevices)
@@ -209,15 +211,19 @@
const size_t numOutputs = params.m_OutputBindings.size();
const size_t containerSize = model.GetOutputSize();
- std::vector<TContainer> outputDataContainers(numOutputs, TContainer(containerSize));
+ // Set up input data container
+ std::vector<TContainer> inputData(1, std::move(inputDataContainer));
+
+ // Set up output data container
+ std::vector<TContainer> outputData(numOutputs, TContainer(containerSize));
// Execute model
- model.Run({ inputDataContainer }, outputDataContainers);
+ model.Run(inputData, outputData);
// Print output tensors
for (size_t i = 0; i < numOutputs; i++)
{
- PrintOutputData(params.m_OutputBindings[i], outputDataContainers[i]);
+ PrintOutputData(params.m_OutputBindings[i], outputData[i]);
}
}
catch (armnn::Exception const& e)