aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorTracy Narine <tracy.narine@arm.com>2023-12-12 13:57:16 +0000
committerTracy Narine <tracy.narine@arm.com>2023-12-13 14:33:14 +0000
commit1685bcca5dae227d90be62b36d66e9897298ce84 (patch)
treef04ae28660fb16fea186245afecfd0f83e918c5e /tests
parent5ea6e3fcc6b0baa822c0b063e474b560ea192339 (diff)
downloadarmnn-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.cpp7
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);