Execute the full export process. This method orchestrates the export process with optional quantization: 1. (Optional) Apply quantization to the model 2. Export the PyTorch model to ExportedProgram 3. Transform and lower to EdgeProgramManager 4. Conv
(self)
| 458 | self._stage_to_artifacts[stage_type] = current_artifact |
| 459 | |
| 460 | def export(self) -> None: |
| 461 | """ |
| 462 | Execute the full export process. |
| 463 | |
| 464 | This method orchestrates the export process with optional quantization: |
| 465 | 1. (Optional) Apply quantization to the model |
| 466 | 2. Export the PyTorch model to ExportedProgram |
| 467 | 3. Transform and lower to EdgeProgramManager |
| 468 | 4. Convert to ExecutorchProgramManager |
| 469 | """ |
| 470 | # Run the pipeline from the beginning |
| 471 | self._run_pipeline() |
| 472 | |
| 473 | def get_stage_artifacts(self) -> Dict[StageType, PipelineArtifact]: |
| 474 | return self._stage_to_artifacts |