diff --git a/src/brevitas/core/zero_point.py b/src/brevitas/core/zero_point.py index db623ef9a..3105d182c 100644 --- a/src/brevitas/core/zero_point.py +++ b/src/brevitas/core/zero_point.py @@ -346,7 +346,7 @@ def forward(self, x: Tensor, scale: Tensor, bit_width: Tensor) -> Tensor: return z -class RuntimeDynamicGroupZeroScaling(brevitas.jit.ScriptModule): +class RuntimeDynamicGroupZeroPoint(brevitas.jit.ScriptModule): def __init__( self,