From 1685bcca5dae227d90be62b36d66e9897298ce84 Mon Sep 17 00:00:00 2001
From: Tracy Narine <tracy.narine@arm.com>
Date: Tue, 12 Dec 2023 13:57:16 +0000
Subject: 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
---
 tests/ExecuteNetwork/ExecuteNetworkParams.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'tests')

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);
-- 
cgit v1.2.1