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
diff --git a/tests/ExecuteNetwork/ExecuteNetworkParams.cpp b/tests/ExecuteNetwork/ExecuteNetworkParams.cpp
index 7a1f68b..4f5a290 100644
--- a/tests/ExecuteNetwork/ExecuteNetworkParams.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetworkParams.cpp
@@ -162,6 +162,13 @@
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);