Release 18.05

1 files changed, 2 insertions, 15 deletions

diff --git a/src/armnn/Graph.cpp b/src/armnn/Graph.cpp
index af3b17ea8b..87bdc2962f 100644
--- a/src/armnn/Graph.cpp
+++ b/src/armnn/Graph.cpp
@@ -3,7 +3,7 @@
 // See LICENSE file in the project root for full license information.
 //
 #include "Graph.hpp"
-#include "Layers.hpp"
+#include "LayersFwd.hpp"
 
 #include <armnn/Utils.hpp>
 #include <armnn/TypesUtils.hpp>
@@ -121,20 +121,7 @@ Status Graph::SerializeToDot(std::ostream& stream)
                 {
                     // Construct the label attribute with HTML markup
                     std::stringstream ss;
-                    {
-                        ss << "< [";
-                        const TensorShape& shape = outputSlot->GetTensorInfo().GetShape();
-                        for (unsigned int i = 0; i < shape.GetNumDimensions(); i++)
-                        {
-                            if (i != 0)
-                            {
-                                ss << ",";
-                            }
-                            ss << shape[i];
-                        }
-                        ss << "] >";
-                    }
-
+                    ss << "< " << outputSlot->GetTensorInfo().GetShape() << " >";
                     edge.GetAttributeSet().AddAttribute("label", ss);
                 }
             }