Searched refs:PrepareCustomConfig (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/torch/ao/quantization/fx/ |
| D | custom_config.py | 42 prepare_custom_config: Optional[PrepareCustomConfig]
46 class PrepareCustomConfig: class
85 prepare_custom_config: Optional[PrepareCustomConfig], argument
87 ) -> PrepareCustomConfig:
105 prepare_custom_config: Optional[PrepareCustomConfig], argument
107 ) -> PrepareCustomConfig:
125 ) -> PrepareCustomConfig:
143 ) -> PrepareCustomConfig:
152 ) -> PrepareCustomConfig:
159 def set_input_quantized_indexes(self, indexes: List[int]) -> PrepareCustomConfig:
[all …]
|
| D | prepare.py | 51 from .custom_config import PrepareCustomConfig, StandaloneModuleConfigEntry
430 prepare_custom_config: PrepareCustomConfig, argument
434 QConfigMapping, Tuple[Any, ...], PrepareCustomConfig, Optional[BackendConfig]
456 prepare_custom_config = config_entry.prepare_custom_config or PrepareCustomConfig()
806 prepare_custom_config: PrepareCustomConfig, argument
968 prepare_custom_config: PrepareCustomConfig, argument
1433 prepare_custom_config: PrepareCustomConfig, argument
1449 prepare_custom_config: PrepareCustomConfig, argument
1915 prepare_custom_config: PrepareCustomConfig, argument
1965 prepare_custom_config: PrepareCustomConfig, argument
[all …]
|
| D | utils.py | 38 from .custom_config import PrepareCustomConfig
78 prepare_custom_config: PrepareCustomConfig
465 prepare_custom_config: PrepareCustomConfig, is_standalone_module: bool argument
|
| D | convert.py | 42 from .custom_config import ConvertCustomConfig, PrepareCustomConfig
1057 prepare_custom_config: PrepareCustomConfig = (
|
| /external/pytorch/torch/ao/quantization/ |
| D | quantize_fx.py | 11 from .fx.custom_config import ConvertCustomConfig, FuseCustomConfig, PrepareCustomConfig
95 prepare_custom_config: Union[PrepareCustomConfig, Dict[str, Any], None] = None, argument
112 prepare_custom_config = PrepareCustomConfig()
123 prepare_custom_config = PrepareCustomConfig.from_dict(prepare_custom_config)
167 prepare_custom_config: Union[PrepareCustomConfig, Dict[str, Any], None] = None, argument
256 prepare_custom_config: Union[PrepareCustomConfig, Dict[str, Any], None] = None, argument
407 prepare_custom_config: Union[PrepareCustomConfig, Dict[str, Any], None] = None, argument
|
| /external/pytorch/torch/ao/quantization/pt2e/ |
| D | prepare.py | 12 from torch.ao.quantization.fx.custom_config import PrepareCustomConfig
563 PrepareCustomConfig(),
|
| /external/pytorch/test/quantization/fx/ |
| D | test_quantize_fx.py | 133 PrepareCustomConfig,
2662 child_prepare_custom_config = PrepareCustomConfig()
2666 prepare_custom_config = PrepareCustomConfig()
2676 child_prepare_custom_config = PrepareCustomConfig()
2680 prepare_custom_config = PrepareCustomConfig()
2689 prepare_custom_config = PrepareCustomConfig()
2700 prepare_custom_config = PrepareCustomConfig()
2706 prepare_custom_config = PrepareCustomConfig()
2713 prepare_custom_config = PrepareCustomConfig()
2719 prepare_custom_config = PrepareCustomConfig()
[all …]
|
| /external/pytorch/docs/source/ |
| D | quantization-support.rst | 120 PrepareCustomConfig
|