swap_module
class torch.quantization.swap_module(mod, mapping, custom_module_class_mapping)
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