finn.transformation.batchnorm_to_affine (module)

class finn.transformation.batchnorm_to_affine.BatchNormToAffine

Bases: finn.transformation.base.Transformation

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

apply(model)