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()的文档
- 返回值
无，模块在原地修改，并附加 qconfig
