Searched refs:StandaloneModuleConfigEntry (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | custom_config.py | 37 class StandaloneModuleConfigEntry: class
67 self.standalone_module_names: Dict[str, StandaloneModuleConfigEntry] = {}
68 self.standalone_module_classes: Dict[Type, StandaloneModuleConfigEntry] = {}
95 self.standalone_module_names[module_name] = StandaloneModuleConfigEntry(
115 self.standalone_module_classes[module_class] = StandaloneModuleConfigEntry(
326 def _make_tuple(key: Any, e: StandaloneModuleConfigEntry): argument
|
| H A D | prepare.py | 51 from .custom_config import PrepareCustomConfig, StandaloneModuleConfigEntry
444 config_entry = StandaloneModuleConfigEntry(None, (), None, None)
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | quantization-support.rst | 122 StandaloneModuleConfigEntry
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 134 StandaloneModuleConfigEntry,
2664 config_entry = StandaloneModuleConfigEntry(
2678 config_entry = StandaloneModuleConfigEntry(
2765 sm_config_entry1 = StandaloneModuleConfigEntry(qm1, ei1, pcc1, bcd1)
2766 sm_config_entry2 = StandaloneModuleConfigEntry(qm2, ei2, pcc2, bcd2)
|