Home
last modified time | relevance | path

Searched refs:additional_fuser_method_mapping (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dfuse_modules.py43 def fuse_known_modules(mod_list, is_qat, additional_fuser_method_mapping=None): argument
59 fuser_method = get_fuser_method(types, additional_fuser_method_mapping)
92 additional_fuser_method_mapping = fuse_custom_config_dict.get(
100 new_mod_list = fuser_func(mod_list, is_qat, additional_fuser_method_mapping)
H A Dfuser_method_mappings.py214 def get_fuser_method(op_list, additional_fuser_method_mapping=None): argument
220 if additional_fuser_method_mapping is None:
221 additional_fuser_method_mapping = {}
223 _DEFAULT_OP_LIST_TO_FUSER_METHOD, additional_fuser_method_mapping