add_quant_dequant

class torch.ao.quantization.add_quant_dequant(module)

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

引數

• module – 具有所有葉子模組 qconfig 屬性的輸入模組

• quantize (that we want to) –

返回

根據 qconfig，子模組被包裝在 QuantWrapper 中的就地修改模組，或者包裝輸入模組的新 QuantWrapper 模組。後一種情況僅發生在輸入模組是葉子模組且我們想對其進行量化時。