Searched refs:custom_module_class_mapping (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | quantize.py | 172 custom_module_class_mapping=None, argument
192 if custom_module_class_mapping is None:
193 custom_module_class_mapping = {}
264 and type_before_parametrizations(child) in custom_module_class_mapping
266 observed_child = custom_module_class_mapping[
273 custom_module_class_mapping[type_before_parametrizations(child)]
283 custom_module_class_mapping,
377 custom_module_class_mapping = prepare_custom_config_dict.get(
402 custom_module_class_mapping=custom_module_class_mapping,
702 custom_module_class_mapping = convert_custom_config_dict.get(
[all …]
|
| H A D | utils.py | 229 custom_module, custom_module_class_mapping, qconfig argument
242 class_mapping = custom_module_class_mapping.get(quant_type, {})
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | convert.py | 895 custom_module_class_mapping: Dict[QuantType, Dict[Type, Type]],
969 observed_custom_module, custom_module_class_mapping, qconfig
1105 custom_module_class_mapping = convert_custom_config.observed_to_quantized_mapping
1233 custom_module_class_mapping,
|
| H A D | prepare.py | 1436 custom_module_class_mapping = prepare_custom_config.float_to_observed_mapping
1438 custom_module, custom_module_class_mapping, qconfig
|