Searched defs:DTypeConfig (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/ |
| H A D | _common_operator_config_utils.py | 135 dtype_configs: List[DTypeConfig],
175 def _get_linear_configs(dtype_configs: List[DTypeConfig]) -> List[BackendPatternConfig]:
474 def _get_cat_config(dtype_configs: List[DTypeConfig]) -> BackendPatternConfig:
482 def _get_ln_configs(dtype_configs: List[DTypeConfig]) -> List[BackendPatternConfig]:
503 dtype_configs: List[DTypeConfig],
550 dtype_configs: List[DTypeConfig],
589 dtype_configs: List[DTypeConfig],
686 def _get_bn_configs(dtype_configs: List[DTypeConfig]) -> List[BackendPatternConfig]:
729 def _get_rnn_op_configs(dtype_configs: List[DTypeConfig]) -> List[BackendPatternConfig]:
751 dtype_configs: List[DTypeConfig],
|
| H A D | backend_config.py | 113 class DTypeConfig: class
512 def add_dtype_config(self, dtype_config: DTypeConfig) -> BackendPatternConfig:
521 self, dtype_configs: List[DTypeConfig]
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | prepare.py | 261 dtype_config: DTypeConfig,
331 dtype_config: DTypeConfig,
|
| H A D | quantize_handler.py | 105 dtype_configs: List[DTypeConfig],
|
| H A D | qconfig_mapping_utils.py | 261 qconfig: QConfig, dtype_configs: List[DTypeConfig]
|