From 2732cca12bac29e1515cee1db5005c73893c61b4 Mon Sep 17 00:00:00 2001
From: Manuel Bottini <manuel.bottini@arm.com>
Date: Tue, 28 May 2019 11:44:41 +0100
Subject: COMPMID-2244: Extend CLFuseBatchNormalization to support
 DepthwiseConvolution weights

Change-Id: I7d1907f35cc4899379073759be2f7cce24e51e9d
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1327
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
---
 arm_compute/core/Types.h | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'arm_compute/core/Types.h')

diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h
index d49315d591..1a49624113 100644
--- a/arm_compute/core/Types.h
+++ b/arm_compute/core/Types.h
@@ -142,6 +142,13 @@ enum class DeconvolutionMethod
     DIRECT, /**< Direct deconvolution */
 };
 
+/** Available FuseBatchNormalizationType*/
+enum class FuseBatchNormalizationType
+{
+    CONVOLUTION,         /**< For Convolution weights */
+    DEPTHWISECONVOLUTION /**< For Depthwise Convolution weights*/
+};
+
 /** Padding mode to use for PadLayer */
 enum class PaddingMode
 {
-- 
cgit v1.2.1