commit 362e03214ceceb95ec30d530ff84e8d9efc198d7
author Rob Hughes <robert.hughes@arm.com> Wed Nov 06 09:36:29 2019 +0000
committer Matteo Martincigh <matteo.martincigh@arm.com> Thu Nov 07 09:34:48 2019 +0000
tree a803126df42d2f06749c9793d88bbec39d7f8f31
parent 5dc816e7d027170992e45d22ae11ae3000de9244

Escape angle brackets in dot file labels

This was a bug that meant invalid dot files were produced due to MemCopy
layers having a name including "->".

Change-Id: If9f5b13d433f6a7328bf0ad8c7ec89cdce2462b0
Signed-off-by: Rob Hughes <robert.hughes@arm.com>

src/armnnUtils/DotSerializer.cpp

diff --git a/src/armnnUtils/DotSerializer.cpp b/src/armnnUtils/DotSerializer.cpp
index b3d3d10..1f36cf7 100644
--- a/src/armnnUtils/DotSerializer.cpp
+++ b/src/armnnUtils/DotSerializer.cpp
@@ -6,6 +6,7 @@
 #include "DotSerializer.hpp"
 
 #include <boost/assert.hpp>
+#include <boost/algorithm/string/replace.hpp>
 #include <sstream>
 #include <cstring>
 
@@ -23,6 +24,14 @@
     }
     return ss.str();
 }
+
+std::string Escape(std::string s)
+{
+    boost::replace_all(s, "<", "\\<");
+    boost::replace_all(s, ">", "\\>");
+    return s;
+}
+
 } //namespace
 
 
@@ -151,7 +160,7 @@
     }
     for (auto & content : m_Contents)
     {
-        ss << content;
+        ss << Escape(content);
         ss << "\\l";
     }
     ss << "}\"";