(
self, config: Optional[EdgeCompileConfig] = None
)
| 505 | |
| 506 | # TODO(ycao): Change this to a composable function. |
| 507 | def to_edge( |
| 508 | self, config: Optional[EdgeCompileConfig] = None |
| 509 | ) -> "ExirExportedProgram": |
| 510 | config = config or EdgeCompileConfig() |
| 511 | assert isinstance( |
| 512 | self.exported_program.graph_module, torch.fx.GraphModule |
| 513 | ), f"type is instead: {type(self.exported_program.graph_module).__name__}" |
| 514 | |
| 515 | return _to_edge(self, config) |
| 516 | |
| 517 | def dump(self) -> None: |
| 518 | print(self.exported_program.graph_module.graph) |