Shortcuts

propagate_qconfig

class torch.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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources