diff options
author | Colm Donelan <colm.donelan@arm.com> | 2023-08-16 21:57:54 +0100 |
---|---|---|
committer | Colm Donelan <colm.donelan@arm.com> | 2024-04-30 15:18:17 +0000 |
commit | 00fe3932fbd6cf8c1181e4943d87ce5144b92b04 (patch) | |
tree | a95fba99c167d518be0908084c7b3ac1606eab91 /tests/ExecuteNetwork/ExecuteNetworkParams.hpp | |
parent | e20aa0fe41e689c4dbbde9adf861d5808840cbe2 (diff) | |
download | armnn-00fe3932fbd6cf8c1181e4943d87ce5144b92b04.tar.gz |
Adding options to serialize networks to ExecuteNetwork.
* Adding serialize-to-armnn parameter to execute network to serialize
the pre-optimized network to a .armnn file and the optimized network
to a .dot file.
Signed-off-by: Colm Donelan <colm.donelan@arm.com>
Change-Id: Ia12cf0b08d36334127264ea3fcc8e64b65f06dae
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetworkParams.hpp')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetworkParams.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetworkParams.hpp b/tests/ExecuteNetwork/ExecuteNetworkParams.hpp index ffcb4f482c..c2bfb951d5 100644 --- a/tests/ExecuteNetwork/ExecuteNetworkParams.hpp +++ b/tests/ExecuteNetwork/ExecuteNetworkParams.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2022, 2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -67,6 +67,7 @@ struct ExecuteNetworkParams std::string m_ComparisonFile; std::vector<armnn::BackendId> m_ComparisonComputeDevices; bool m_CompareWithTflite; + bool m_SerializeToArmNN; // Ensures that the parameters for ExecuteNetwork fit together void ValidateParams(); |