Searched refs:model_to_quantize (Results 1 – 4 of 4) sorted by relevance
21 model_to_quantize = copy.deepcopy(float_model) variable23 model_to_quantize.eval()
20 model_to_quantize = copy.deepcopy(float_model) variable22 model_to_quantize.eval()
456 model_to_quantize = copy.deepcopy(model_fp)457 model_to_quantize.eval()462 model_prepared = quantize_fx.prepare_fx(model_to_quantize, qconfig_mapping, example_inputs)471 model_to_quantize = copy.deepcopy(model_fp)473 model_to_quantize.eval()475 model_prepared = quantize_fx.prepare_fx(model_to_quantize, qconfig_mapping, example_inputs)484 model_to_quantize = copy.deepcopy(model_fp)486 model_to_quantize.train()488 model_prepared = quantize_fx.prepare_qat_fx(model_to_quantize, qconfig_mapping, example_inputs)496 model_to_quantize = copy.deepcopy(model_fp)[all …]
111 exported_model = export_for_training(model_to_quantize, example_inputs).module()