IVGCVSW-6786 Add import if memory aligned option to ExecuteNetwork
Change-Id: Ib038e7b2616195a64715e3a7126da1368bbca1d3
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp
index 1f19584..92a2946 100644
--- a/tests/ExecuteNetwork/ExecuteNetwork.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp
@@ -405,6 +405,7 @@
inferenceModelParams.m_ThreadPoolSize = params.m_ThreadPoolSize;
inferenceModelParams.m_OutputDetailsToStdOut = params.m_OutputDetailsToStdOut;
inferenceModelParams.m_OutputDetailsOnlyToStdOut = params.m_OutputDetailsOnlyToStdOut;
+ inferenceModelParams.m_ImportInputsIfAligned = params.m_ImportInputsIfAligned;
for(const std::string& inputName: params.m_InputNames)
{
diff --git a/tests/ExecuteNetwork/ExecuteNetworkParams.hpp b/tests/ExecuteNetwork/ExecuteNetworkParams.hpp
index cb8c2fb..deaf55f 100644
--- a/tests/ExecuteNetwork/ExecuteNetworkParams.hpp
+++ b/tests/ExecuteNetwork/ExecuteNetworkParams.hpp
@@ -65,6 +65,7 @@
std::string m_MLGOTuningFilePath;
TfLiteExecutor m_TfLiteExecutor;
size_t m_ThreadPoolSize;
+ bool m_ImportInputsIfAligned;
// Ensures that the parameters for ExecuteNetwork fit together
void ValidateParams();
diff --git a/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp b/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp
index 8ee66cf..681dc8a 100644
--- a/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp
@@ -423,8 +423,12 @@
("output-network-details-only",
"Outputs layer tensor infos and descriptors to std out without profiling events. Defaults to off.",
cxxopts::value<bool>(m_ExNetParams.m_OutputDetailsOnlyToStdOut)->default_value("false")
- ->implicit_value("true"));
+ ->implicit_value("true"))
+ ("import-inputs-if-aligned",
+ "In & Out tensors will be imported per inference if the memory alignment allows. Defaults to false.",
+ cxxopts::value<bool>(m_ExNetParams.m_ImportInputsIfAligned)->default_value("false")
+ ->implicit_value("true"));
}
catch (const std::exception& e)
{