Home
last modified time | relevance | path

Searched refs:set_module_name_regex (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dqconfig_mapping.py261 def set_module_name_regex( member in QConfigMapping
372 conf.set_module_name_regex(module_name_regex, qconfig)
/aosp_15_r20/external/pytorch/torch/ao/ns/fx/
H A Dqconfig_multi_mapping.py159 def set_module_name_regex( member in QConfigMultiMapping
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_numeric_suite_fx.py2321 .set_module_name_regex("fc", [torch.ao.quantization.default_qconfig])
2358 .set_module_name_regex("fc", [torch.ao.quantization.default_qconfig])
H A Dtest_quantize_fx.py2461 qconfig_mapping.set_module_name_regex("foo.*bar", qconfig1)
2462 qconfig_mapping.set_module_name_regex("foo.*", qconfig2)
2467 qconfig_mapping.set_module_name_regex("foo.*bar", qconfig3)
2602 .set_module_name_regex("foo.*bar", qconfig1) \
2603 .set_module_name_regex("foo.*", qconfig2) \