| /aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/ |
| H A D | onednn.py | 76 def _fuse_linear_bn_leaky_relu(is_qat, linear, bn, leaky_relu): argument
134 def _fuse_conv_add_left(is_qat, add, conv, _): argument
158 def _fuse_conv_bn_add_left(is_qat, add, bn_conv, _): argument
220 def _fuse_conv_add_right(is_qat, add, _, conv): argument
244 def _fuse_conv_bn_add_right(is_qat, add, _, bn_conv): argument
318 def _fuse_conv_add_relu_left(is_qat, relu, add_pattern): argument
347 def _fuse_conv_bn_add_relu_left(is_qat, relu, add_pattern): argument
414 def _fuse_conv_add_relu_right(is_qat, relu, add_pattern): argument
443 def _fuse_conv_bn_add_relu_right(is_qat, relu, add_pattern): argument
|
| /aosp_15_r20/external/executorch/backends/mediatek/quantizer/ |
| H A D | qconfig.py | 67 is_qat, argument
94 dtype, is_symmetric, is_per_channel, is_qat, quant_min=None, quant_max=None argument
125 def get_a16w16_quant_config(is_per_channel, is_qat) -> QuantizationConfig: argument
134 def get_a16w8_quant_config(is_per_channel, is_qat) -> QuantizationConfig: argument
143 def get_a16w4_quant_config(is_per_channel, is_qat) -> QuantizationConfig: argument
154 def get_a8w8_quant_config(is_per_channel, is_qat) -> QuantizationConfig: argument
163 def get_a8w4_quant_config(is_per_channel, is_qat) -> QuantizationConfig: argument
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | fuser_method_mappings.py | 20 def fuse_conv_bn(is_qat, conv, bn): argument
64 def fuse_conv_bn_relu(is_qat, conv, bn, relu): argument
119 def fuse_linear_bn(is_qat, linear, bn): argument
153 def fuse_convtranspose_bn(is_qat, convt, bn): argument
188 def fuser_method(is_qat, m1, m2): argument
231 def reversed(is_qat, x, y): argument
238 def reversed(is_qat, x, w): argument
|
| H A D | fuse_modules.py | 43 def fuse_known_modules(mod_list, is_qat, additional_fuser_method_mapping=None): argument
86 is_qat, argument
110 is_qat, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_x86inductor_quantizer.py | 546 is_qat=False, argument
1218 is_qat=False, argument
1359 def _test_linear_binary_helper(self, is_qat=False, is_dynamic=False): argument
1532 def _test_linear_binary_unary_helper(self, is_qat=False, is_dynamic=False): argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_mkldnn_pattern_matcher.py | 76 def get_default_quantizer(is_qat, is_dynamic): argument
133 self, mod, inputs, is_qat=False, is_dynamic=False, quantizer=None argument
164 is_qat=False, argument
1440 is_qat=False, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_fuse_eager.py | 372 def custom_fuse_func(module, is_qat, add_fuser_mapping): argument
|
| /aosp_15_r20/external/executorch/backends/qualcomm/quantizer/ |
| H A D | quantizer.py | 170 self, quant_dtype: QuantDtype, is_qat=False, act_observer=None argument
|
| /aosp_15_r20/external/executorch/examples/qualcomm/ |
| H A D | utils.py | 232 is_qat=False, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/quantizer/ |
| H A D | xnnpack_quantizer_utils.py | 57 is_qat: bool = False variable in QuantizationConfig
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 1248 is_qat=False, argument
2942 mod, inputs, is_qat=False, is_dynamic=False, quantizer=None argument
2945 def get_default_quantizer(is_qat, is_dynamic): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | prepare.py | 369 is_qat, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 746 def fuse_conv_bn_relu(is_qat, relu, add_pattern): argument
793 def fuse_conv_bn_relu(is_qat, relu, add_pattern): argument
862 def fuse_conv_relu(is_qat, conv, relu): argument
865 def fuse_conv_res_relu(is_qat, relu, add_pattern): argument
|