add_quant_dequant#

class torch.ao.quantization.add_quant_dequant(module)[source]#

Wrap the leaf child module in QuantWrapper if it has a valid qconfig Note that this function will modify the children of module inplace and it can return a new module which wraps the input module as well.

Parameters

• module – input module with qconfig attributes for all the leaf modules

• quantize (that we want to) –

Returns

Either the inplace modified module with submodules wrapped in QuantWrapper based on qconfig or a new QuantWrapper module which wraps the input module, the latter case only happens when the input module is a leaf module and we want to quantize it.