Home
last modified time | relevance | path

Searched defs:BackendConfig (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/
H A Dutils.py30 backend_config: BackendConfig,
38 def get_qat_module_classes(backend_config: BackendConfig) -> Tuple[type, ...]:
46 def get_fused_module_classes(backend_config: BackendConfig) -> Tuple[type, ...]:
55 backend_config: BackendConfig,
64 backend_config: BackendConfig,
77 backend_config: BackendConfig,
91 backend_config: BackendConfig,
101 backend_config: BackendConfig,
120 backend_config: BackendConfig,
H A Dbackend_config.py289 class BackendConfig: class
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dquantize_fx.py77 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
97 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
168 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
207 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
258 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
408 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
516 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
560 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
627 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
676 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dprepare.py262 backend_config: BackendConfig,
392 backend_config: BackendConfig,
432 parent_backend_config: Optional[BackendConfig],
510 backend_config: BackendConfig,
565 backend_config: BackendConfig,
809 backend_config: Optional[BackendConfig] = None,
971 backend_config: Optional[BackendConfig] = None,
1451 backend_config: BackendConfig,
1916 backend_config: BackendConfig,
1990 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
H A Dconvert.py538 def _run_weight_observers(observed: GraphModule, backend_config: BackendConfig) -> None:
667 backend_config: Optional[BackendConfig],
731 backend_config: BackendConfig,
985 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
H A Dlstm_utils.py22 backend_config: Optional[BackendConfig] = None,
145 backend_config: Optional[BackendConfig] = None,
H A Dcustom_config.py86 backend_config: Optional[BackendConfig],
106 backend_config: Optional[BackendConfig],
H A Dfuse.py36 backend_config: Union[BackendConfig, Dict[str, Any], None] = None,
H A Dfuse_handler.py125 backend_config: BackendConfig,
H A Dquantize_handler.py142 backend_config: BackendConfig,
H A Dqconfig_mapping_utils.py387 qconfig_mapping: QConfigMapping, backend_config: BackendConfig
/aosp_15_r20/external/pytorch/torch/ao/ns/
H A D_numeric_suite_fx.py854 backend_config: BackendConfig,
963 backend_config: BackendConfig,
1044 backend_config: BackendConfig,
/aosp_15_r20/external/pytorch/torch/distributed/
H A Ddistributed_c10d.py350 class BackendConfig: class
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py5776 backend_config: BackendConfig,