propagate_qconfig

函数torch.ao.quantization.propagate_qconfig_(module, qconfig_dict=None, prepare_custom_config_dict=None)[源代码]

将 qconfig 逐层传递并通过模块层次结构分配，在每个叶子模块上设置 qconfig 属性

参数

• module – 输入模块

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

• prepare_custom_config_dict - 用于模块自定义处理的字典，详情请参阅 prepare_fx() 的文档。

返回值

无需单独操作，模块会直接在原地进行修改，并附加上 qconfig