diff options
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetworkParams.cpp')
-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); |