Home
last modified time | relevance | path

Searched defs:qconfig_mapping (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dqconfig_mapping_utils.py313 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 Dtest_numeric_suite_fx.py2565 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 Dtest_quantize_fx.py6288 self, module, example_inputs, qconfig_mapping, argument