diff --git a/src/finn/transformation/streamline/reorder.py b/src/finn/transformation/streamline/reorder.py index b47f269dd6f2671c3d98c9316954483c0e72f14f..f4c1dc1306b67e5807c25cfb08c961729dbfbdf6 100644 --- a/src/finn/transformation/streamline/reorder.py +++ b/src/finn/transformation/streamline/reorder.py @@ -533,7 +533,7 @@ class MoveScalarLinearPastInvariants(Transformation): if prod0 is None: continue - if prod0.op_type == "Mul" or prod0.op_type == "Add": + if prod0.op_type in ["Mul", "Add", "Div"]: # check if second input of producer is an initializer init0 = model.get_initializer(prod0.input[1]) # if either initializer is None, skip