propagate_qconfig

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

通过模块层级传播 qconfig，并在每个叶子模块上分配 `qconfig` 属性

参数

• module – 输入模块

• qconfig_dict – 一个字典，它将子模块的名称或类型映射到量化配置。除非为子模块指定了 qconfig（当子模块已具有 qconfig 属性时），否则 qconfig 将应用于给定模块的所有子模块。

• prepare_custom_config_dict – 用于自定义模块处理的字典，请参阅 `prepare_fx()` 函数的文档。

返回

None，模块被就地修改并附加了 qconfig