propagate_qconfig

class torch.ao.quantization.propagate_qconfig_(module, qconfig_dict=None, prepare_custom_config_dict=None) [исходный код]

Распространяет qconfig по иерархии модулей и назначает атрибут qconfig каждому листовому модулю

Параметры:

• module – входной модуль
• qconfig_dict – словарь, сопоставляющий имя или тип подмодуля с конфигурацией квантования; qconfig применяется ко всем подмодулям заданного модуля, если для подмодулей не указаны собственные qconfig (когда у подмодуля уже есть атрибут qconfig)
• prepare_custom_config_dict – словарь для пользовательской обработки модулей; см. документацию по prepare_fx()

Возвращает:

None; модуль изменяется на месте, к нему добавляется qconfig