| /aosp_15_r20/external/executorch/kernels/quantized/cpu/ |
| H A D | op_quantize.cpp | 34 int64_t quant_max, in check_quantize_per_tensor_args()
96 int64_t quant_max) { in quantize_val()
113 int64_t quant_max, in quantize_per_tensor_out()
170 int64_t quant_max, in quantize_per_tensor_tensor_args_out()
214 int64_t quant_max, in quantize_per_tensor_tensor_args_out()
230 int64_t quant_max, in quantize_per_tensor_out()
246 int64_t quant_max, in quantize_per_channel_out()
368 int64_t quant_max, in quantize_per_channel_out()
386 int64_t quant_max, in quantize_per_token_out()
435 int64_t quant_max, in quantize_per_token_out()
|
| H A D | op_dequantize.cpp | 34 int64_t quant_max, in check_dequantize_per_tensor_args()
82 int64_t quant_max, in dequantize_per_tensor_out()
142 int64_t quant_max, in dequantize_per_tensor_tensor_args_out()
194 int64_t quant_max, in dequantize_per_channel_out()
340 int64_t quant_max, in dequantize_per_channel_out()
368 int64_t quant_max, in dequantize_per_tensor_out()
385 int64_t quant_max, in dequantize_per_tensor_tensor_args_out()
401 int64_t quant_max, in dequantize_per_token_out()
460 int64_t quant_max, in dequantize_per_token_out()
|
| H A D | op_choose_qparams.cpp | 223 int64_t quant_max, in choose_qparams_tensor_out()
239 int64_t quant_max, in choose_qparams_tensor_out()
257 int64_t quant_max = 127; in choose_qparams_per_token_asymmetric_out() local
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/representation/ |
| H A D | rewrite.py | 402 quant_max, argument
428 quant_max, argument
465 quant_max, argument
490 quant_max, argument
599 def _quantize_per_tensor_int8(x_fp32, scale, zero_point, quant_min, quant_max): argument
607 x_fp32, scale, zero_point, quant_min, quant_max argument
631 def _dequantize_per_tensor_int8(x_i8, scale, zero_point, quant_min, quant_max): argument
639 x_i8, scale, zero_point, quant_min, quant_max argument
662 x_fp32, scales, zero_points, ch_axis, quant_min, quant_max argument
671 x_fp32, scales, zero_points, ch_axis, quant_min, quant_max argument
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/ |
| H A D | FakeQuantPerTensorAffine.cpp | 36 int64_t quant_max) { in fake_quantize_per_tensor_affine()
47 int64_t quant_max) { in fake_quantize_per_tensor_affine()
74 int64_t quant_max) { in fake_quantize_per_tensor_affine_cachemask()
98 int64_t quant_max) { in _fake_quantize_per_tensor_affine_cachemask_tensor_qparams()
139 int64_t quant_max, in _get_zero_point_from_tensor()
153 int64_t quant_max, in _fake_quantize_learnable_per_tensor_affine()
167 int64_t quant_max, in _fake_quantize_learnable_per_tensor_affine_backward()
|
| H A D | FakeQuantPerChannelAffine.cpp | 38 int64_t quant_max) { in fake_quantize_per_channel_affine()
50 int64_t quant_max) { in fake_quantize_per_channel_affine_cachemask()
136 int64_t quant_max) { in _get_rounded_zero_point()
147 int64_t quant_max, in _fake_quantize_learnable_per_channel_affine()
161 int64_t quant_max, in _fake_quantize_learnable_per_channel_affine_backward()
|
| /aosp_15_r20/external/executorch/kernels/quantized/test/ |
| H A D | op_embedding_test.cpp | 45 int64_t quant_max = 255; in test_dtype() local
107 int64_t quant_max = 255; in TEST() local
180 int64_t quant_max = 255; in TEST() local
237 int64_t quant_max = 255; in TEST() local
267 int64_t quant_max = 255; in TEST() local
297 int64_t quant_max = 255; in TEST() local
326 int64_t quant_max = 255; in TEST() local
355 int64_t quant_max = 255; in TEST() local
|
| H A D | op_add_test.cpp | 47 int64_t quant_max = 255; in test_dtype() local
115 int64_t quant_max = 255; in TEST() local
187 int64_t quant_max = 255; in TEST() local
304 int64_t quant_max = 255; in TEST() local
367 int64_t quant_max = 255; in TEST() local
|
| H A D | quantized_ops_aot_register.cpp | 36 int64_t quant_max, in quantize_per_token_out_no_context()
51 int64_t quant_max, in quantize_per_token_aten()
77 int64_t quant_max, in dequantize_per_token_out_no_context()
101 int64_t quant_max, in dequantize_per_token_aten()
|
| H A D | op_dequantize_test.cpp | 41 int64_t quant_max = 255; in test_dtype() local
78 int64_t quant_max = 255; in TEST() local
107 int64_t quant_max = 255; in TEST() local
136 int64_t quant_max = 255; in TEST() local
|
| H A D | op_embedding2b_test.cpp | 36 int64_t quant_max = 1; in TEST() local
114 int64_t quant_max = 1; in TEST() local
142 int64_t quant_max = 1; in TEST() local
170 int64_t quant_max = 1; in TEST() local
|
| H A D | op_quantize_test.cpp | 40 int64_t quant_max = 127; in test_dtype() local
70 int64_t quant_max = 255; in TEST() local
102 int64_t quant_max = 127; in TEST() local
132 int64_t quant_max = 255; in TEST() local
|
| H A D | op_embedding4b_test.cpp | 36 int64_t quant_max = 7; in TEST() local
111 int64_t quant_max = 7; in TEST() local
138 int64_t quant_max = 7; in TEST() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/ |
| H A D | FakeQuantizeCore.cu | 29 int64_t quant_max) { in fake_quantize_tensor_cachemask_kernel_cuda()
62 int64_t quant_max) { in fake_quantize_tensor_cachemask_tensor_qparams_kernel_cuda()
98 int64_t quant_max, in _fake_quantize_grad_learnable_tensor_kernel_cuda()
130 const int64_t quant_max in _fake_quant_per_channel_cachemask_cuda_helper()
184 TensorIterator &iter, TensorIterator &iter_mask, int64_t quant_min, int64_t quant_max) { in fake_quant_per_channel_cachemask_cuda()
190 …rnable_channel_kernel_cuda(TensorIterator &iter, int64_t quant_min, int64_t quant_max, float grad_… in _fake_quantize_grad_learnable_channel_kernel_cuda()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| H A D | quantize_and_dequantize_test.cc | 42 const int quant_max = (1 << num_bits) - 1; in TEST() local
73 const int quant_max = (1 << num_bits) - 1; in TEST() local
104 const int quant_max = (1 << num_bits) - 1; in TEST() local
135 const int quant_max = (1 << num_bits) - 1; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | quantize_and_dequantize_test_util.cc | 37 const int quant_max = (1 << num_bits) - 1; in QuantAndDequant_Dim2Bits8Test() local
74 const int quant_max = (1 << num_bits) - 1; in QuantAndDequant_Dim3Bits8_NegativeRangeTest() local
110 const int quant_max = (1 << num_bits) - 1; in QuantAndDequant_Dim3Bits16Test() local
147 const int quant_max = (1 << num_bits) - 1; in QuantAndDequant_Dim2Bits16_NegativeRangeTest() local
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | observer.py | 219 quant_max=None, argument
485 quant_max=None, argument
612 quant_max=None, argument
695 quant_max=None, argument
902 quant_max=None, argument
991 quant_max=None, argument
1403 quant_max=255, argument
1456 quant_max=None, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/quantizer/ |
| H A D | quantizer.py | 44 quant_max: Optional[int] = None variable in QuantizationSpec
73 quant_max: Optional[int] = None variable in FixedQParamsQuantizationSpec
106 quant_max: Optional[int] = None variable in DerivedQuantizationSpec
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_workflow_ops.py | 37 def _fake_quantize_per_tensor_affine_reference(X, scale, zero_point, quant_min, quant_max): argument
44 def _fake_quantize_per_tensor_affine_grad_reference(dY, X, scale, zero_point, quant_min, quant_max): argument
53 …learnable_per_tensor_affine_grad_reference(dY, X, scale, zero_point, quant_min, quant_max, device): argument
93 def _quantize_per_tensor(x, scale, zero_point, quant_min, quant_max): argument
98 dY, X, per_channel_scale, per_channel_zero_point, axis, quant_min, quant_max, device): argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cpu_repro.py | 924 x, scale, zero_point, use_dequant, use_quant, quant_min, quant_max, dtype argument
988 x, scale, zero_point, use_dequant, use_quant, quant_min, quant_max, dtype argument
1053 def fn(x, scale, zero_point, quant_min, quant_max, dtype): argument
1112 quant_max, argument
1197 def fn(input, scales, zero_points, quant_min, quant_max, dtype): argument
1236 def fn(input, scales, zero_points, axis, quant_min, quant_max, dtype): argument
1333 quant_max, argument
1390 x, groups, output_scale, output_zero_point, quant_min, quant_max, dtype argument
1436 quant_max, argument
1810 def fq(input, scales, zero_points, axis, quant_min, quant_max): argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantization_utils_test.cc | 83 const int quant_max = 127; in TEST_F() local
107 const int quant_max = 127; in TEST_F() local
130 const int quant_max = 127; in TEST_F() local
151 const int quant_max = 127; in TEST_F() local
173 const int quant_max = 127; in TEST_F() local
|
| /aosp_15_r20/external/executorch/backends/mediatek/quantizer/ |
| H A D | qconfig.py | 70 quant_max=None, argument
94 dtype, is_symmetric, is_per_channel, is_qat, quant_min=None, quant_max=None argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | fake_quantize_ops.cc | 30 const float quant_max, float* nudged_min, float* nudged_max, in CpuNudge()
60 xla::XlaOp quant_max = in XlaNudge() local
101 const float quant_max = (1 << num_bits) - 1; in FakeQuantWithMinMaxArgsGradOp() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantized.py | 196 …channel_affine_reference(X, per_channel_scale, per_channel_zero_point, axis, quant_min, quant_max): argument
210 …ffine_grad_reference(dY, X, per_channel_scale, per_channel_zero_point, axis, quant_min, quant_max): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | fused_obs_fake_quant.cpp | 154 const int64_t quant_max, in fused_moving_avg_obs_fake_quant_cpu()
232 const int64_t quant_max, in fused_moving_avg_obs_fake_quant()
|