qonnx.transformation.batchnorm_to_affine (module)

class qonnx.transformation.batchnorm_to_affine.BatchNormToAffine

Bases: qonnx.transformation.base.Transformation

Replaces any test-time BatchNorm layers with Mul-Add layers.

apply(model)