Home
last modified time | relevance | path

Searched defs:quant_max (Results 1 – 25 of 51) sorted by relevance

123

/external/pytorch/aten/src/ATen/native/quantized/
DFakeQuantPerTensorAffine.cpp36 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()
DFakeQuantPerChannelAffine.cpp38 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()
/external/pytorch/aten/src/ATen/native/quantized/cuda/
DFakeQuantizeCore.cu29 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()
/external/executorch/kernels/quantized/cpu/
Dop_quantize.cpp34 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()
Dop_dequantize.cpp34 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()
Dop_choose_qparams.cpp223 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
/external/executorch/kernels/quantized/test/
Dop_embedding_test.cpp45 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
Dquantized_ops_aot_register.cpp36 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()
Dop_dequantize_test.cpp41 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
Dop_embedding2b_test.cpp36 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
Dop_quantize_test.cpp40 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
Dop_embedding4b_test.cpp36 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
Dop_add_test.cpp47 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
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dquantize_and_dequantize_test.cc42 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
/external/pytorch/torch/ao/quantization/pt2e/representation/
Drewrite.py402 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 …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dquantize_and_dequantize_test_util.cc37 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
/external/pytorch/torch/ao/quantization/quantizer/
Dquantizer.py44 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
/external/executorch/backends/mediatek/quantizer/
Dqconfig.py70 quant_max=None, argument
94 dtype, is_symmetric, is_per_channel, is_qat, quant_min=None, quant_max=None argument
/external/pytorch/torch/ao/quantization/
Dobserver.py219 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
/external/pytorch/test/quantization/core/
Dtest_workflow_ops.py37 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
/external/executorch/backends/cadence/reference/operators/
Dquantize_per_tensor.cpp28 int64_t quant_max, in quantize_per_tensor_out()
Ddequantize_per_tensor.cpp26 int64_t quant_max, in dequantize_per_tensor_out()
/external/pytorch/aten/src/ATen/native/quantized/cpu/
Dfused_obs_fake_quant.cpp154 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()
/external/pytorch/torch/testing/_internal/
Dcommon_quantized.py196 …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
/external/executorch/backends/cadence/hifi/operators/
Dquantize_per_tensor.cpp30 __ET_UNUSED int64_t quant_max, in quantize_per_tensor_out()

123