Recipe and Input class definitions with e2e export (#10034)
Based on the discussion in https://github.com/pytorch/executorch/discussions/9027 This PR adds the executorch.export API and all the supporting components required for it. At a high level the executorch.export API takes in a model, example inputs and a recipe, then underneath the hood executes all the steps required to export, quantize and lower the model based on the recipe. The pipeline consists of a staged setup where each major step in the process such as Export, Quantization etc. is listed as a separate stage and a chain of these is formed and then executed. The result of this will be that we'll get a PTE file which can then be executed on device. The major new components added in this PR are: - class definition for ExportSession and ExportRecipe - Definitions for each stage in the process - executorch.export API which will return a session object that the user can then use to get access to the PTE file, run the model via pybindings, print delegation info etc.
T
Tarun Karuturi committed
1b593ad1943e53c33de4445f0aab483d181bf81e
Parent: d67fb52
Committed by GitHub <noreply@github.com>
on 5/15/2025, 12:48:38 AM