From e2220551b7a64b929650ba9a60529c31e70c13c5 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 20 Jul 2018 13:23:44 +0100 Subject: COMPMID-1367: Enable NHWC in graph examples Change-Id: Iabc54a3a1bdcd46a9a921cda39c7c85fef672b72 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/141449 Reviewed-by: Giorgio Arena Reviewed-by: Anthony Barbier Tested-by: Jenkins --- src/graph/printers/DotGraphPrinter.cpp | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'src/graph/printers/DotGraphPrinter.cpp') diff --git a/src/graph/printers/DotGraphPrinter.cpp b/src/graph/printers/DotGraphPrinter.cpp index 61cf42356f..ef156ea252 100644 --- a/src/graph/printers/DotGraphPrinter.cpp +++ b/src/graph/printers/DotGraphPrinter.cpp @@ -47,17 +47,19 @@ void DotGraphVisitor::visit(BatchNormalizationLayerNode &n) _info = ss.str(); } -void DotGraphVisitor::visit(ConvolutionLayerNode &n) +void DotGraphVisitor::visit(ConcatenateLayerNode &n) { std::stringstream ss; - ss << n.convolution_method(); + ss << "Enabled: " << n.is_enabled(); + ss << R"( \n )"; + ss << "Axis: " << n.concatenation_axis(); _info = ss.str(); } -void DotGraphVisitor::visit(DepthConcatenateLayerNode &n) +void DotGraphVisitor::visit(ConvolutionLayerNode &n) { std::stringstream ss; - ss << "Enabled: " << n.is_enabled(); + ss << n.convolution_method(); _info = ss.str(); } -- cgit v1.2.1