get_swapped_custom_module_class

class torch.ao.quantization.utils.get_swapped_custom_module_class(custom_module, custom_module_class_mapping, qconfig) [исходный код]

Получить класс наблюдаемого/квантованного пользовательского модуля, на который необходимо заменить custom_module.

Входные данные:

• custom_module: входной объект, может быть экземпляром модуля с плавающей точкой или наблюдаемого пользовательского модуля
• custom_module_class_mapping: сопоставление класса модуля с плавающей точкой с классом наблюдаемого модуля или класса наблюдаемого модуля с классом квантованного модуля
• qconfig: конфигурация qconfig для пользовательского модуля

Выходные данные:

Соответствующий класс наблюдаемого/квантованного пользовательского модуля для экземпляра входного пользовательского модуля.