get_custom_module_class_keys

class torch.ao.quantization.fx.utils.get_custom_module_class_keys(custom_module_mapping) [source]

Получить все уникальные ключи пользовательских модулей в словаре пользовательской конфигурации.

Пример входных данных:

{
    QuantType.STATIC: {CustomModule1: ObservedCustomModule},
    QuantType.DYNAMIC: {CustomModule2: DynamicObservedCustomModule},
    QuantType.WEIGHT_ONLY: {CustomModule3: WeightOnlyObservedCustomModule},
}

Пример выходных данных:

# extract the keys across all inner STATIC, DYNAMIC, and WEIGHT_ONLY dicts
[CustomModule1, CustomModule2, CustomModule3]

Тип возвращаемого значения:

list[Any]