Method __init__

W8ASpike/quant_linear.py:22–27

(self, in_features: int, out_features: int, bias: bool = True, device=None, dtype=None, w_group_size=128, dynamic_sfr=3.0)

class QuantLinear(nn.Linear):
    def __init__(self, in_features: int, out_features: int, bias: bool = True, device=None, dtype=None, w_group_size=128, dynamic_sfr=3.0):
        super().__init__(in_features, out_features, bias, device=device, dtype=dtype)

        self.k = dynamic_sfr
        self.w_group_size = w_group_size
        self.weight_quantizer = Quantizer(in_features, out_features, w_group_size)

    def forward(self, x):
        # BLD