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
