(self, tensor)
| 41 | self.calib_algo = self.kwargs.get("calib_algo", "minmax") |
| 42 | |
| 43 | def get_tensor_range(self, tensor): |
| 44 | if self.calib_algo == "minmax": |
| 45 | return self.get_minmax_range(tensor) |
| 46 | elif self.calib_algo == "mse": |
| 47 | return self.get_mse_range(tensor) |
| 48 | else: |
| 49 | raise ValueError(f"Unsupported calibration algorithm: {self.calib_algo}") |
| 50 | |
| 51 | def get_minmax_range(self, tensor): |
| 52 | if self.granularity == "per_tensor": |
no test coverage detected