propagate_qconfig¶
-
class
torch.quantization.propagate_qconfig_(module, qconfig_dict=None, allow_list=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)
allow_list – a set that lists out allowable modules to be propagated with qconfig
- Returns
None, module is modified inplace with qconfig attached
