diff options
Diffstat (limited to 'tests/ExecuteNetwork')
| -rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 1 | ||||
| -rw-r--r-- | tests/ExecuteNetwork/ExecuteNetworkParams.hpp | 1 | ||||
| -rw-r--r-- | tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp | 6 |
3 files changed, 7 insertions, 1 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 1f19584c68..92a2946847 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -405,6 +405,7 @@ int MainImpl(const ExecuteNetworkParams& params, 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 cb8c2fb386..deaf55f6e5 100644 --- a/tests/ExecuteNetwork/ExecuteNetworkParams.hpp +++ b/tests/ExecuteNetwork/ExecuteNetworkParams.hpp @@ -65,6 +65,7 @@ struct ExecuteNetworkParams 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 8ee66cf64b..681dc8a611 100644 --- a/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp +++ b/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp @@ -423,8 +423,12 @@ ProgramOptions::ProgramOptions() : m_CxxOptions{"ExecuteNetwork", ("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) { |