diff --git a/src/finn/transformation/streamline/reorder.py b/src/finn/transformation/streamline/reorder.py index 2cde191435894c72cadf73af82df3f315fb2998c..452366607943db77d7ef3764c67b0fdd4b0fbb40 100644 --- a/src/finn/transformation/streamline/reorder.py +++ b/src/finn/transformation/streamline/reorder.py @@ -560,7 +560,9 @@ class MoveScalarLinearPastInvariants(Transformation): graph_modified = True else: continue - model = model.transform(InferShapes()) + if graph_modified: + model = model.transform(InferShapes()) + model = model.transform(InferDataTypes()) return (model, graph_modified)