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/nodes/Nodes.h | 1 + 1 file changed, 1 insertion(+) (limited to 'arm_compute/graph/nodes/Nodes.h') diff --git a/arm_compute/graph/nodes/Nodes.h b/arm_compute/graph/nodes/Nodes.h index c891bc2ca2..52e2f88528 100644 --- a/arm_compute/graph/nodes/Nodes.h +++ b/arm_compute/graph/nodes/Nodes.h @@ -39,6 +39,7 @@ #include "arm_compute/graph/nodes/FlattenLayerNode.h" #include "arm_compute/graph/nodes/FullyConnectedLayerNode.h" #include "arm_compute/graph/nodes/FusedConvolutionBatchNormalizationNode.h" +#include "arm_compute/graph/nodes/FusedDepthwiseConvolutionBatchNormalizationNode.h" #include "arm_compute/graph/nodes/GenerateProposalsLayerNode.h" #include "arm_compute/graph/nodes/InputNode.h" #include "arm_compute/graph/nodes/NormalizationLayerNode.h" -- cgit v1.2.1