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)
|