Searched defs:qconfig_mapping (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | qconfig_mapping_utils.py | 313 def _get_module_name_regex_qconfig(qconfig_mapping, module_name, fallback_qconfig): argument
321 def _get_module_name_qconfig(qconfig_mapping, module_name, fallback_qconfig): argument
333 qconfig_mapping, module_type, module_name, global_qconfig argument
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_numeric_suite_fx.py | 2565 def _test_extract_weights_impl(self, m, example_input, qconfig_mapping): argument
2611 def _test_add_loggers_impl(self, m, example_input, qconfig_mapping): argument
|
| H A D | test_quantize_fx.py | 6288 self, module, example_inputs, qconfig_mapping, argument
|