propagate_qconfig

class torch.ao.quantization.propagate_qconfig_(module, qconfig_dict=None, prepare_custom_config_dict=None) [source]

Propagate qconfig through the module hierarchy and assign qconfig attribute on each leaf module

Parameters

• module – input module
• qconfig_dict – dictionary that maps from name or type of submodule to quantization configuration, qconfig applies to all submodules of a given module unless qconfig for the submodules are specified (when the submodule already has qconfig attribute)
• prepare_custom_config_dict – dictionary for custom handling of modules see docs for prepare_fx()

Returns

None, module is modified inplace with qconfig attached