please use torch.ao.quantization.quantize since torch.quantization.quantize is going to be deprecated
torch.ao.quantization.quantize
torch.quantization.quantize