add_quant_dequant

class torch.ao.quantization.add_quant_dequant(module)

如果葉子子模組具有有效的 qconfig，則將其包裝在 QuantWrapper 中。請注意，此函數將就地修改模組的子模組，並且它可以返回一個也包裝輸入模組的新模組。

參數

• module – 輸入模組，其中所有葉子模組都有 qconfig 屬性

• quantize (我們想要) –

返回

根據 qconfig 包裝在 QuantWrapper 中的就地修改模組，或者包裝輸入模組的新 QuantWrapper 模組。後者僅在輸入模組是葉子模組並且我們想要量化它時發生。