Correct zero_point_impl inj

Xilinx · Oct 1, 2024 · 7e1fd5e · 7e1fd5e
1 parent eb6e108
commit 7e1fd5e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/brevitas/quant/base.py b/src/brevitas/quant/base.py
@@ -481,6 +481,10 @@ class AccumulatorAwareZeroCenterWeightQuant(AccumulatorAwareWeightQuant):
     """
     per_channel_pre_norm = AccumulatorAwareZeroCenterPerChannelPreNorm
 
+    @value
+    def pre_zero_point_impl():
+        return this.per_channel_pre_norm.pre_zero_point_impl
+
 
 class MSESubInjectorBase(ExtendedInjector):