diff --git a/src/finn/builder/build_dataflow_steps.py b/src/finn/builder/build_dataflow_steps.py index 1c1861e5286e92abcf983056f8263daae14334e8..4c9b076b6acb1e725421d5067ac74ad42c3dbafb 100644 --- a/src/finn/builder/build_dataflow_steps.py +++ b/src/finn/builder/build_dataflow_steps.py @@ -158,6 +158,7 @@ def step_streamline(model: ModelWrapper, cfg: DataflowBuildConfig): topologies. """ + model = model.transform(absorb.AbsorbSignBiasIntoMultiThreshold()) model = model.transform(MoveScalarLinearPastInvariants()) model = model.transform(Streamline()) need_lowering = len(model.get_nodes_by_op_type("Conv")) > 0