diff options
| -rw-r--r-- | tests/ExecuteNetwork/ArmNNExecutor.cpp | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/tests/ExecuteNetwork/ArmNNExecutor.cpp b/tests/ExecuteNetwork/ArmNNExecutor.cpp index 943d3aad07..9657f05b6d 100644 --- a/tests/ExecuteNetwork/ArmNNExecutor.cpp +++ b/tests/ExecuteNetwork/ArmNNExecutor.cpp @@ -26,7 +26,6 @@ ArmNNExecutor::ArmNNExecutor(const ExecuteNetworkParams& params, armnn::IRuntime auto optNet = OptimizeNetwork(network.get()); m_IOInfo = GetIOInfo(optNet.get()); - SetupInputsAndOutputs(); armnn::ProfilingDetailsMethod profilingDetailsMethod = ProfilingDetailsMethod::Undefined; if (params.m_OutputDetailsOnlyToStdOut) @@ -57,6 +56,8 @@ ArmNNExecutor::ArmNNExecutor(const ExecuteNetworkParams& params, armnn::IRuntime return; } + SetupInputsAndOutputs(); + if (m_Params.m_Iterations > 1) { std::stringstream msg; @@ -415,11 +416,12 @@ void ArmNNExecutor::SetupInputsAndOutputs() } } - if (m_Params.m_ImportInputsIfAligned) - { - m_ImportedInputIds.push_back( - m_Runtime->ImportInputs(m_NetworkId, m_InputTensorsVec.back(), armnn::MemorySource::Malloc)); - } + } + + if (m_Params.m_ImportInputsIfAligned) + { + m_ImportedInputIds.push_back( + m_Runtime->ImportInputs(m_NetworkId, inputTensors, armnn::MemorySource::Malloc)); } m_InputTensorsVec.emplace_back(inputTensors); } |