Searched refs:quantize_linear_modules (Results 1 – 2 of 2) sorted by relevance
146 fb_int8 = torch.jit.quantized.quantize_linear_modules(fb)151 fb_fp16 = torch.jit.quantized.quantize_linear_modules(fb, torch.float16)
89 def quantize_linear_modules(module, dtype=torch.int8): function