Searched refs:captured_model (Results 1 – 2 of 2) sorted by relevance
/external/executorch/examples/qualcomm/ |
D | utils.py | 187 def ptq_calibrate(captured_model, quantizer, dataset): argument 188 annotated_model = prepare_pt2e(captured_model, quantizer) 199 def qat_train(ori_model, captured_model, quantizer, dataset): argument 202 prepare_qat_pt2e(captured_model, quantizer) 260 captured_model = torch.export.export(model, inputs).module() 267 model, captured_model, quantizer, qat_training_data 272 annotated_model = ptq_calibrate(captured_model, quantizer, dataset)
|
/external/executorch/examples/mediatek/aot_utils/oss_utils/ |
D | utils.py | 33 captured_model = torch.export.export_for_training(model, inputs).module() 34 annotated_model = prepare_pt2e(captured_model, quantizer)
|