(self, bit, symmetric, granularity, **kwargs)
| 32 | |
| 33 | class BaseQuantizer(object): |
| 34 | def __init__(self, bit, symmetric, granularity, **kwargs): |
| 35 | self.bit = bit |
| 36 | self.sym = symmetric |
| 37 | self.granularity = granularity |
| 38 | self.kwargs = kwargs |
| 39 | if self.granularity == "per_group": |
| 40 | self.group_size = self.kwargs["group_size"] |
| 41 | self.calib_algo = self.kwargs.get("calib_algo", "minmax") |
| 42 | |
| 43 | def get_tensor_range(self, tensor): |
| 44 | if self.calib_algo == "minmax": |