ArmNN
 24.08
FuseBatchNorm.hpp File Reference
Include dependency graph for FuseBatchNorm.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  FuseBatchNorm< ConvLayer, ArmnnType, T >
 

Namespaces

 armnn
 Copyright (c) 2021 ARM Limited and Contributors.
 
 armnn::optimizations
 

Typedefs

using FuseBatchNormIntoConvolution2DFloat32 = OptimizeForExclusiveConnection< Convolution2dLayer, BatchNormalizationLayer, FuseBatchNorm< Convolution2dLayer, armnn::DataType::Float32 > >
 
using FuseBatchNormIntoConvolution2DFloat16 = OptimizeForExclusiveConnection< Convolution2dLayer, BatchNormalizationLayer, FuseBatchNorm< Convolution2dLayer, armnn::DataType::Float16 > >
 
using FuseBatchNormIntoDepthwiseConvolution2DFloat32 = OptimizeForExclusiveConnection< DepthwiseConvolution2dLayer, BatchNormalizationLayer, FuseBatchNorm< DepthwiseConvolution2dLayer, armnn::DataType::Float32 > >
 
using FuseBatchNormIntoDepthwiseConvolution2DFloat16 = OptimizeForExclusiveConnection< DepthwiseConvolution2dLayer, BatchNormalizationLayer, FuseBatchNorm< DepthwiseConvolution2dLayer, armnn::DataType::Float16 > >