From bffb41e06c1276af00e1605ef934d05fa61f7127 Mon Sep 17 00:00:00 2001
From: Manuel Bottini <manuel.bottini@arm.com>
Date: Thu, 20 Jun 2019 16:00:27 +0100
Subject: COMPMID-2273: Fuse Batch Normalization with Depthwise Convolution
 layer at graph level (only for CL)

Change-Id: I1d941c6e66722f39583bf68148c980bb28ff89a1
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1423
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
---
 src/graph/printers/DotGraphPrinter.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/graph/printers/DotGraphPrinter.cpp')

diff --git a/src/graph/printers/DotGraphPrinter.cpp b/src/graph/printers/DotGraphPrinter.cpp
index c939de1b64..46f6ee828e 100644
--- a/src/graph/printers/DotGraphPrinter.cpp
+++ b/src/graph/printers/DotGraphPrinter.cpp
@@ -85,6 +85,14 @@ void DotGraphVisitor::visit(FusedConvolutionBatchNormalizationNode &n)
     _info = ss.str();
 }
 
+void DotGraphVisitor::visit(FusedDepthwiseConvolutionBatchNormalizationNode &n)
+{
+    ARM_COMPUTE_UNUSED(n);
+    std::stringstream ss;
+    ss << "FusedDepthwiseConvolutionBatchNormalizationNode";
+    _info = ss.str();
+}
+
 void DotGraphVisitor::visit(NormalizationLayerNode &n)
 {
     std::stringstream ss;
-- 
cgit v1.2.1