diff options
Diffstat (limited to 'tests/ExecuteNetwork')
-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 |