| /aosp_15_r20/external/pytorch/torch/ao/ns/fx/ |
| H A D | qconfig_multi_mapping.py | 26 def _remove_duplicates_and_none(qconfig_list: List[QConfigAny]) -> None:
80 self, qconfig_list: List[QConfigAny], style: str
128 qconfig_list: List[QConfigAny],
141 def set_global(self, global_qconfig_list: List[QConfigAny]) -> QConfigMultiMapping:
150 self, object_type: Union[Callable, str], qconfig_list: List[QConfigAny]
160 self, module_name_regex: str, qconfig_list: List[QConfigAny]
172 self, module_name: str, qconfig_list: List[QConfigAny]
186 qconfig_list: List[QConfigAny],
|
| H A D | n_shadows_utils.py | 457 list_of_node_name_to_qconfig: List[Dict[str, QConfigAny]], argument
617 list_of_node_name_to_qconfig: List[Dict[str, QConfigAny]], argument
717 node_name_to_qconfig: Dict[str, QConfigAny], argument
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | _lower_to_native_backend.py | 238 def should_skip_lowering(op: torch.fx.node.Node, qconfig_map: Dict[str, QConfigAny]): argument
520 qconfig_map: Dict[str, QConfigAny], argument
598 qconfig_map: Dict[str, QConfigAny], argument
655 model: GraphModule, qconfig_map: Dict[str, QConfigAny] argument
709 model: GraphModule, qconfig_map: Dict[str, QConfigAny] argument
846 model: GraphModule, qconfig_map: Dict[str, QConfigAny] argument
946 model: GraphModule, qconfig_map: Dict[str, QConfigAny] argument
1077 def _lower_quantized_binary_op(model: GraphModule, qconfig_map: Dict[str, QConfigAny]): argument
1301 qconfig_map: Dict[str, QConfigAny], argument
|
| H A D | prepare.py | 260 qconfig: QConfigAny,
330 qconfig: QConfigAny,
391 qconfig: QConfigAny,
431 parent_qconfig: QConfigAny,
508 qconfig: QConfigAny,
562 qconfig: QConfigAny,
801 qconfig: QConfigAny,
963 qconfig: QConfigAny,
1431 qconfig: QConfigAny,
1448 node_name_to_qconfig: Dict[str, QConfigAny], argument
[all …]
|
| H A D | convert.py | 95 node_name_to_qconfig: Dict[str, QConfigAny], argument
355 node_name_to_qconfig: Dict[str, QConfigAny], argument
526 node: Argument, node_name_to_qconfig: Dict[str, QConfigAny] argument
584 node_name_to_qconfig: Dict[str, QConfigAny], argument
730 node_name_to_qconfig: Dict[str, QConfigAny], argument
|
| H A D | utils.py | 489 qconfig: QConfigAny = None,
511 qconfig: QConfigAny = None,
842 qconfig: QConfigAny,
|
| H A D | qconfig_mapping_utils.py | 36 fallback_qconfig: QConfigAny,
308 fallback_qconfig: QConfigAny,
|
| H A D | lower_to_fbgemm.py | 14 qconfig_map: Dict[str, QConfigAny], argument
|
| H A D | lower_to_qnnpack.py | 14 qconfig_map: Dict[str, QConfigAny], argument
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | qconfig_mapping.py | 244 def set_global(self, global_qconfig: QConfigAny) -> QConfigMapping:
252 self, object_type: Union[Callable, str], qconfig: QConfigAny
262 self, module_name_regex: str, qconfig: QConfigAny
284 def set_module_name(self, module_name: str, qconfig: QConfigAny) -> QConfigMapping:
293 self, module_name: str, object_type: Callable, index: int, qconfig: QConfigAny
|
| H A D | qconfig.py | 565 QConfigAny = Optional[QConfig] variable
570 qconfig: QConfigAny, module: Optional[nn.Module]
654 def qconfig_equals(q1: QConfigAny, q2: QConfigAny):
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/ |
| H A D | prepare.py | 315 qconfig: QConfigAny,
398 qconfig: QConfigAny,
|