Skip to content

Commit 7e1fd5e

Browse files
committed
Correct zero_point_impl inj
1 parent eb6e108 commit 7e1fd5e

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/brevitas/quant/base.py

+4
Original file line numberDiff line numberDiff line change
@@ -481,6 +481,10 @@ class AccumulatorAwareZeroCenterWeightQuant(AccumulatorAwareWeightQuant):
481481
"""
482482
per_channel_pre_norm = AccumulatorAwareZeroCenterPerChannelPreNorm
483483

484+
@value
485+
def pre_zero_point_impl():
486+
return this.per_channel_pre_norm.pre_zero_point_impl
487+
484488

485489
class MSESubInjectorBase(ExtendedInjector):
486490

0 commit comments

Comments
 (0)