qonnx.transformation.batchnorm_to_affine (module)
- class qonnx.transformation.batchnorm_to_affine.BatchNormToAffine
Bases:
qonnx.transformation.base.TransformationReplaces any test-time BatchNorm layers with Mul-Add layers.
- apply(model)
Bases: qonnx.transformation.base.Transformation
Replaces any test-time BatchNorm layers with Mul-Add layers.