diff options
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 { |