more fixes

Xilinx · Jan 23, 2025 · d391858 · d391858
1 parent b7c8776
commit d391858
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/brevitas/graph/equalize.py b/src/brevitas/graph/equalize.py
@@ -1300,6 +1300,8 @@ def apply(self, alpha):
                     co_optimize_act_weights=self.co_optimize_act_weights)
             scale_factors.append(scale_factor_region)
             rewriters.append(rewriters_region)
+            for r in rewriters_region:
+                r.apply(self.model)
         return scale_factors, rewriters_region
 
     def insert_mul_node(self, scale, shape, axis, act_node, batch_dim=0):