aboutsummaryrefslogtreecommitdiff
path: root/Utils.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'Utils.hpp')
-rw-r--r--Utils.hpp3
1 files changed, 3 insertions, 0 deletions
diff --git a/Utils.hpp b/Utils.hpp
index 49b8b8d8..e6b56be0 100644
--- a/Utils.hpp
+++ b/Utils.hpp
@@ -48,4 +48,7 @@ void DumpTensor(const std::string& dumpDir,
const std::string& tensorName,
const armnn::ConstTensor& tensor);
+void ExportNetworkGraphToDotFile(const armnn::IOptimizedNetwork& optimizedNetwork,
+ const std::string& dumpDir,
+ const Model& model);
}