diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2019-02-21 16:29:43 +0000 |
---|---|---|
committer | derek.lamberti <derek.lamberti@arm.com> | 2019-02-22 09:57:45 +0000 |
commit | 8ddae33ada66cc2ecdc0cde7799d81dda7532fb5 (patch) | |
tree | 3f835fe51cd00ddd1660e431b29db65991c66b06 /tests/InferenceModel.hpp | |
parent | c64ea9fdf975a65e9a1dd67b44469add270d6f8b (diff) | |
download | armnn-8ddae33ada66cc2ecdc0cde7799d81dda7532fb5.tar.gz |
IVGSVSW-2736 Fix dangling reference to temporary storage
Change-Id: Ie6c553798eac732f37148d81970366e5a4ede1be
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'tests/InferenceModel.hpp')
-rw-r--r-- | tests/InferenceModel.hpp | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index 3b3dd95e88..6e73f5275f 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -203,14 +203,16 @@ public: for (const std::string& inputLayerName : params.m_InputBindings) { - BindingPointInfo inputBinding = parser->GetNetworkInputBindingInfo(subGraphId, inputLayerName); - inputBindings.push_back(inputBinding); + armnnDeserializer::BindingPointInfo inputBinding = + parser->GetNetworkInputBindingInfo(subGraphId, inputLayerName); + inputBindings.push_back(std::make_pair(inputBinding.m_BindingId, inputBinding.m_TensorInfo)); } for (const std::string& outputLayerName : params.m_OutputBindings) { - BindingPointInfo outputBinding = parser->GetNetworkOutputBindingInfo(subGraphId, outputLayerName); - outputBindings.push_back(outputBinding); + armnnDeserializer::BindingPointInfo outputBinding = + parser->GetNetworkOutputBindingInfo(subGraphId, outputLayerName); + outputBindings.push_back(std::make_pair(outputBinding.m_BindingId, outputBinding.m_TensorInfo)); } return network; |