From 93f5f97e40070cad5beb7ccf9c959684d0e93684 Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Thu, 31 Jan 2019 13:12:34 +0000 Subject: IVGCVSW-2601 Fix for ExecuteNetwork output data printing bug Change-Id: I7e8d377cdbddf0176ae05d8b6b2fd03c6f614168 Signed-off-by: Aron Virginas-Tar --- tests/ExecuteNetwork/ExecuteNetwork.cpp | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) (limited to 'tests') diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index e6e6cdf80d..29780104c2 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -140,9 +140,11 @@ std::vector ParseArray(std::istream& stream) void PrintOutputData(const std::string& outputLayerName, const std::vector& data) { std::cout << outputLayerName << ": "; - std::copy(data.begin(), data.end(), - std::ostream_iterator(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& computeDevices) @@ -209,15 +211,19 @@ int MainImpl(const char* modelPath, const size_t numOutputs = params.m_OutputBindings.size(); const size_t containerSize = model.GetOutputSize(); - std::vector outputDataContainers(numOutputs, TContainer(containerSize)); + // Set up input data container + std::vector inputData(1, std::move(inputDataContainer)); + + // Set up output data container + std::vector 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) -- cgit v1.2.1