Save the model to a .pte file. Args: output_name (Optional[str]): The name of the .pte file.
(self, output_name: str)
| 577 | return self.get_executorch_program_manager().buffer |
| 578 | |
| 579 | def save_to_pte(self, output_name: str) -> None: |
| 580 | """ |
| 581 | Save the model to a .pte file. |
| 582 | |
| 583 | Args: |
| 584 | output_name (Optional[str]): The name of the .pte file. |
| 585 | """ |
| 586 | assert output_name, "Need a valid output name" |
| 587 | save_pte_program(self.get_executorch_program_manager(), output_name) |
| 588 | |
| 589 | def get_example_input( |
| 590 | self, method_name: str = "forward" |