Home
last modified time | relevance | path

Searched refs:captured_model (Results 1 – 2 of 2) sorted by relevance

/external/executorch/examples/qualcomm/
Dutils.py187 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/
Dutils.py33 captured_model = torch.export.export_for_training(model, inputs).module()
34 annotated_model = prepare_pt2e(captured_model, quantizer)