code: def quantize_module(model, orig_class, quantize_fn):
      policy = {orig_class: quantize_fn}
      return _quantize_module(model, policy)
  
imports: import torch
label_desc: Use image node_id 2 for calling a global function with example usage: quantize_module(model, orig_class, quantize_fn) and returns: _quantize_module