swap_module

class torch.quantization.swap_module(mod, mapping, custom_module_class_mapping) [source]

Swaps the module if it has a quantized counterpart and it has an observer attached.

Parameters:

• mod – input module
• mapping – a dictionary that maps from nn module to nnq module

Returns:

The corresponding quantized module of mod