Home
last modified time | relevance | path

Searched refs:StandaloneModuleConfigEntry (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dcustom_config.py37 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 Dprepare.py51 from .custom_config import PrepareCustomConfig, StandaloneModuleConfigEntry
444 config_entry = StandaloneModuleConfigEntry(None, (), None, None)
/aosp_15_r20/external/pytorch/docs/source/
H A Dquantization-support.rst122 StandaloneModuleConfigEntry
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py134 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)