diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-01-30 17:41:47 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-01-31 10:13:39 +0000 |
commit | 9b93747c4b2513d71f6ee975cdff9672ee421ad0 (patch) | |
tree | 2d08f8253bbcf9ec2f029511cf7c589b6411ed51 | |
parent | b5f7df527e374ac0aaeabc51ab9029d4a40748c9 (diff) | |
download | armnn-9b93747c4b2513d71f6ee975cdff9672ee421ad0.tar.gz |
IVGCVSW-2564 Fix output buffer initialization bug in ExecuteNetwork
Change-Id: I69eac52c9c944a5d209c5c7432c1b7f46e64f05b
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 9 |
1 files 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<TParser, TDataType> model(params, runtime); - // Executes the model - const size_t numOutputs = params.m_OutputBindings.size(); - std::vector<TContainer> outputDataContainers(numOutputs); + const size_t numOutputs = params.m_OutputBindings.size(); + const size_t containerSize = model.GetOutputSize(); + + std::vector<TContainer> outputDataContainers(numOutputs, TContainer(containerSize)); + + // Execute model model.Run({ inputDataContainer }, outputDataContainers); // Print output tensors |