Searched refs:fuser_method (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/pytorch/torch/ao/quantization/ |
H A D | fuser_method_mappings.py | 188 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 D | fuse_modules.py | 59 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 D | utils.py | 81 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 D | backend_config.py | 451 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 D | onednn.py | 529 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 D | fuse_handler.py | 109 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 D | README.md | 99 …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 D | test_backend_config.py | 206 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)
|