(self, export_stage: Optional[Export] = None)
| 213 | ) |
| 214 | |
| 215 | def export(self, export_stage: Optional[Export] = None): |
| 216 | return self._run_stage( |
| 217 | export_stage |
| 218 | or self._get_default_stage( |
| 219 | StageType.EXPORT, dynamic_shapes=self.dynamic_shapes |
| 220 | ), |
| 221 | self.example_inputs, |
| 222 | ) |
| 223 | |
| 224 | def to_edge(self, to_edge_stage: Optional[ToEdge] = None): |
| 225 | if not to_edge_stage: |
no test coverage detected