Method forward

quantize/quantizer.py:77–82

(self, x: torch.Tensor)

    

    def forward(self, x: torch.Tensor):
        if self.n_bits >= 16 or not self.enable:
            return x

        x_dequant = self.fake_quant(x)
        return x_dequant