From 9b93747c4b2513d71f6ee975cdff9672ee421ad0 Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Wed, 30 Jan 2019 17:41:47 +0000 Subject: IVGCVSW-2564 Fix output buffer initialization bug in ExecuteNetwork Change-Id: I69eac52c9c944a5d209c5c7432c1b7f46e64f05b Signed-off-by: Aron Virginas-Tar --- tests/ExecuteNetwork/ExecuteNetwork.cpp | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index d783a0e2cf..e6e6cdf80d 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -206,9 +206,12 @@ int MainImpl(const char* modelPath, params.m_SubgraphId = subgraphId; InferenceModel model(params, runtime); - // Executes the model - const size_t numOutputs = params.m_OutputBindings.size(); - std::vector outputDataContainers(numOutputs); + const size_t numOutputs = params.m_OutputBindings.size(); + const size_t containerSize = model.GetOutputSize(); + + std::vector outputDataContainers(numOutputs, TContainer(containerSize)); + + // Execute model model.Run({ inputDataContainer }, outputDataContainers); // Print output tensors -- cgit v1.2.1