diff options
| author | Manuel Bottini <manuel.bottini@arm.com> | 2019-05-28 11:44:41 +0100 |
|---|---|---|
| committer | Manuel Bottini <manuel.bottini@arm.com> | 2019-06-13 16:01:42 +0000 |
| commit | 2732cca12bac29e1515cee1db5005c73893c61b4 (patch) | |
| tree | 050d4c20b51b2b642be21512f9b4a900e18ce88c /arm_compute/core/Types.h | |
| parent | b3a0a60d0b570c58d84324059abb5caceae2561c (diff) | |
| download | ComputeLibrary-2732cca12bac29e1515cee1db5005c73893c61b4.tar.gz | |
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>
Diffstat (limited to 'arm_compute/core/Types.h')
| -rw-r--r-- | arm_compute/core/Types.h | 7 |
1 files changed, 7 insertions, 0 deletions
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 { |