diff options
author | Tracy Narine <tracy.narine@arm.com> | 2023-12-12 13:57:16 +0000 |
---|---|---|
committer | Tracy Narine <tracy.narine@arm.com> | 2023-12-13 14:33:14 +0000 |
commit | 1685bcca5dae227d90be62b36d66e9897298ce84 (patch) | |
tree | f04ae28660fb16fea186245afecfd0f83e918c5e /tests | |
parent | 5ea6e3fcc6b0baa822c0b063e474b560ea192339 (diff) | |
download | armnn-1685bcca5dae227d90be62b36d66e9897298ce84.tar.gz |
MLCE-1168 ExecuteNetwork option --visualize-optimized-model does not work with "--tflite-executor parser" option
* Allow serialize to dot for the parser delegate
Signed-off-by: Tracy Narine <tracy.narine@arm.com>
Change-Id: Id9edef143f0e0cc02dd7b1bfe87d36035bf44c6d
Diffstat (limited to 'tests')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetworkParams.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetworkParams.cpp b/tests/ExecuteNetwork/ExecuteNetworkParams.cpp index 7a1f68bcaf..4f5a290a0d 100644 --- a/tests/ExecuteNetwork/ExecuteNetworkParams.cpp +++ b/tests/ExecuteNetwork/ExecuteNetworkParams.cpp @@ -162,6 +162,13 @@ armnnDelegate::DelegateOptions ExecuteNetworkParams::ToDelegateOptions() const optimizerOptions.AddModelOption(networkOption); } + if (m_EnableLayerDetails) + { + fs::path filename = m_ModelPath; + filename.replace_extension("dot"); + delegateOptions.SetSerializeToDot(filename); + } + { armnn::BackendOptions option("GpuAcc", {{"FastMathEnabled", m_EnableFastMath}}); optimizerOptions.AddModelOption(option); |