Home
last modified time | relevance | path

Searched refs:set_backend_pattern_config (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/
H A Dbackend_config.py374 def set_backend_pattern_config(self, config: BackendPatternConfig) -> BackendConfig: member in BackendConfig
394 self.set_backend_pattern_config(conf)
417 conf.set_backend_pattern_config(d)
419 conf.set_backend_pattern_config(BackendPatternConfig.from_dict(d))
H A Dtensorrt.py81 .set_backend_pattern_config(addmm_config)
82 .set_backend_pattern_config(cat_config)
H A Dnative.py149 .set_backend_pattern_config(_get_cat_config(default_op_dtype_configs))
199 .set_backend_pattern_config(_get_cat_config(default_op_dtype_configs))
H A DREADME.md105 .set_backend_pattern_config(linear_config) \
106 .set_backend_pattern_config(conv_relu_config) \
107 .set_backend_pattern_config(fused_conv_relu_config)
H A Dfbgemm.py113 .set_backend_pattern_config(_get_cat_config(default_op_dtype_configs))
H A Dx86.py110 .set_backend_pattern_config(_get_cat_config(default_op_dtype_configs))
H A Dqnnpack.py158 .set_backend_pattern_config(_get_cat_config(default_op_dtype_configs))
H A Donednn.py633 .set_backend_pattern_config(_get_cat_config(default_op_dtype_configs))
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_backend_config.py286 conf.set_backend_pattern_config(backend_op_config1)
290 conf.set_backend_pattern_config(backend_op_config2)
320 … conf = BackendConfig("name1").set_backend_pattern_config(op1).set_backend_pattern_config(op2)
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py758 .set_backend_pattern_config(conv_bn_res_relu_config1) \
759 .set_backend_pattern_config(conv_bn_res_relu_config2)
818 backend_config = BackendConfig().set_backend_pattern_config(conv_bn_res_relu_config)
885 .set_backend_pattern_config(conv_relu_config) \
886 .set_backend_pattern_config(conv_res_relu_config)
5836 .set_backend_pattern_config(BackendPatternConfig(torch.nn.Linear)
5888 .set_backend_pattern_config(BackendPatternConfig(torch.nn.Linear)
6054 .set_backend_pattern_config(backend_pattern_config)