tensorplay.ao.quantization.quantize

tensorplay.ao.quantization.quantize(model, run_fn, run_args, mapping=None, inplace=False)[source]

Prepare, calibrate through run_fn(*run_args) and convert.

Only submodules carrying a qconfig are quantized; attach qconfigs (or pass a qconfig_dict through propagate_qconfig_()) before calling.