(self, tensor)
| 95 | return tensor, scales, zeros, qmax, qmin |
| 96 | |
| 97 | def fake_quant_tensor(self, tensor): |
| 98 | org_shape = tensor.shape |
| 99 | org_dtype = tensor.dtype |
| 100 | tensor, scales, zeros, qmax, qmin = self.get_tensor_qparams(tensor) |
| 101 | tensor = self.quant_dequant(tensor, scales, zeros, qmax, qmin) |
| 102 | tensor = self.restore_tensor(tensor, org_shape).to(org_dtype) |
| 103 | return tensor |
| 104 | |
| 105 | def real_quant_tensor(self, tensor): |
| 106 | org_shape = tensor.shape |
no test coverage detected