Searched refs:_calculate_dynamic_qparams (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | hypothesis_utils.py | 16 from torch.testing._internal.common_quantized import _calculate_dynamic_qparams, _calculate_dynamic…
212 scale, zp = _calculate_dynamic_qparams(X, qparams[2])
|
| H A D | common_quantized.py | 51 def _calculate_dynamic_qparams(X, dtype, reduce_range=False, qscheme=torch.per_tensor_affine): function
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_tensor.py | 28 def _calculate_dynamic_qparams(X, dtype, reduce_range=False): function
221 scale, zero_pt = _calculate_dynamic_qparams(mat2quant, dtype, reduce_flag)
1367 X_scale, X_zp = _calculate_dynamic_qparams(X, torch.quint8, reduce_range=reduce_range)
1485 scale, zero_point = _calculate_dynamic_qparams(X, qdtype)
1497 scale, zero_point = _calculate_dynamic_qparams(X, torch.quint8)
1515 scale, zero_point = _calculate_dynamic_qparams(X, qdtype)
|
| H A D | test_quantized_op.py | 28 …rch.testing._internal.common_quantized import _quantize, _dequantize, _calculate_dynamic_qparams, \
2938 scale, zero_point = _calculate_dynamic_qparams(x, dtype=dtype)
3051 scale, zero_point = _calculate_dynamic_qparams(
3210 X_scale, X_zp = _calculate_dynamic_qparams(X_fp32, torch.quint8, reduce_range)
3283 W_scale, W_zp = _calculate_dynamic_qparams(W_fp32, torch.qint8)
3288 X_scale, X_zp = _calculate_dynamic_qparams(X_fp32, torch.quint8)
3437 s, z = _calculate_dynamic_qparams(X, torch.quint8, reduce_range)
3449 s, z = _calculate_dynamic_qparams(H, torch.quint8, reduce_range)
3451 s, z = _calculate_dynamic_qparams(C, torch.quint8, reduce_range)
3680 s, z = _calculate_dynamic_qparams(X_fp32, dtype, reduce_range)
[all …]
|
| H A D | test_quantized_module.py | 28 _calculate_dynamic_qparams,
1430 s, z = _calculate_dynamic_qparams(X_fp32, dtype, reduce_range)
1607 W_scale, W_zp = _calculate_dynamic_qparams(W, torch.qint8, qscheme=qscheme)
|