Home
last modified time | relevance | path

Searched refs:fuser_method (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dfuser_method_mappings.py188 def fuser_method(is_qat, m1, m2): function
191 return fuser_method
225 fuser_method = all_mappings.get(op_list, None)
226 assert fuser_method is not None, f"did not find fuser method for: {op_list} "
227 return fuser_method
288 fuser_method = None
290 fuser_method = fuser_method_mapping.get(op_pattern, None)
291 if fuser_method is not None:
293 assert fuser_method is not None, f"did not find fuser method for: {op_pattern} "
294 return fuser_method
H A Dfuse_modules.py59 fuser_method = get_fuser_method(types, additional_fuser_method_mapping)
60 if fuser_method is None:
63 fused = fuser_method(is_qat, *mod_list)
/aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/
H A Dutils.py81 if config.fuser_method is not None:
85 fuser_method = _get_fuser_method_in_reversed_nested_tuple_format(config)
86 fuser_method_mapping[pattern] = fuser_method
300 assert config.fuser_method is not None
302 return config.fuser_method
308 return _reverse2(config.fuser_method)
310 return _reverse3(config.fuser_method)
H A Dbackend_config.py451 self.fuser_method: Optional[Callable] = None
576 def set_fuser_method(self, fuser_method: Callable) -> BackendPatternConfig:
591 self.fuser_method = fuser_method
727 if self.fuser_method is not None:
728 backend_pattern_config_dict[FUSER_METHOD_DICT_KEY] = self.fuser_method
H A Donednn.py529 fuser_method, argument
538 .set_fuser_method(fuser_method)
545 .set_fuser_method(fuser_method)
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dfuse_handler.py109 fuser_method = get_fuser_method_new(matched_module_types, fuser_method_mapping)
112 fused_module = fuser_method(is_qat, *matched_modules)
129 if config.fuser_method is not None:
H A DREADME.md99 …dule in the pattern, like Linear), with a fused module by calling the fuser_method for this patter…
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_backend_config.py206 self.assertTrue(conf.fuser_method is None)
208 self.assertEqual(conf.fuser_method, self._fuser_method)
243 self.assertEqual(conf1.fuser_method, self._fuser_method)
257 self.assertTrue(conf2.fuser_method is None)