From 362e03214ceceb95ec30d530ff84e8d9efc198d7 Mon Sep 17 00:00:00 2001
From: Rob Hughes <robert.hughes@arm.com>
Date: Wed, 6 Nov 2019 09:36:29 +0000
Subject: 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 | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/armnnUtils/DotSerializer.cpp b/src/armnnUtils/DotSerializer.cpp
index b3d3d103af..1f36cf73ae 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 @@ std::string Indent(int numSpaces)
     }
     return ss.str();
 }
+
+std::string Escape(std::string s)
+{
+    boost::replace_all(s, "<", "\\<");
+    boost::replace_all(s, ">", "\\>");
+    return s;
+}
+
 } //namespace
 
 
@@ -151,7 +160,7 @@ NodeContent::~NodeContent()
     }
     for (auto & content : m_Contents)
     {
-        ss << content;
+        ss << Escape(content);
         ss << "\\l";
     }
     ss << "}\"";
-- 
cgit v1.2.1