From bffb41e06c1276af00e1605ef934d05fa61f7127 Mon Sep 17 00:00:00 2001 From: Manuel Bottini 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 Reviewed-on: https://review.mlplatform.org/c/1423 Comments-Addressed: Arm Jenkins Reviewed-by: Michele Di Giorgio Tested-by: Arm Jenkins --- arm_compute/graph/printers/DotGraphPrinter.h | 1 + 1 file changed, 1 insertion(+) (limited to 'arm_compute/graph/printers/DotGraphPrinter.h') diff --git a/arm_compute/graph/printers/DotGraphPrinter.h b/arm_compute/graph/printers/DotGraphPrinter.h index 9d2ea46fde..c28a17b21a 100644 --- a/arm_compute/graph/printers/DotGraphPrinter.h +++ b/arm_compute/graph/printers/DotGraphPrinter.h @@ -57,6 +57,7 @@ public: void visit(DepthwiseConvolutionLayerNode &n) override; void visit(EltwiseLayerNode &n) override; void visit(FusedConvolutionBatchNormalizationNode &n) override; + void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) override; void visit(NormalizationLayerNode &n) override; void visit(PoolingLayerNode &n) override; void default_visit() override; -- cgit v1.2.1