propagate_qconfig#

classtorch.ao.quantization.propagate_qconfig_(module,qconfig_dict=None,prepare_custom_config_dict=None)[source]#

Propagate qconfig through the module hierarchy and assignqconfigattribute on each leaf module

Parameters

• module – input module

• qconfig_dict – dictionary that maps from name or type of submodule toquantization configuration, qconfig applies to all submodules of agiven module unless qconfig for the submodules are specified (whenthe submodule already has qconfig attribute)

• prepare_custom_config_dict – dictionary for custom handling of modulessee docs forprepare_fx()

Returns

None, module is modified inplace with qconfig attached