propagate_qconfig¶
- class torch.ao.quantization.propagate_qconfig_(module, qconfig_dict=None, prepare_custom_config_dict=None)[source][source]¶
在模組層次結構中傳播 qconfig，並在每個葉子模組上分配 qconfig 屬性
- 引數
module – 輸入模組
qconfig_dict – 將子模組名稱或型別對映到量化配置的字典，qconfig 應用於給定模組的所有子模組，除非已為子模組指定了 qconfig（當子模組已具有 qconfig 屬性時）
prepare_custom_config_dict – 用於模組自定義處理的字典，請參閱
prepare_fx()的文件
- 返回值
None，模組將就地修改並附帶 qconfig
