(self, graph_module: torch.fx.GraphModule)
| 1355 | ) |
| 1356 | |
| 1357 | def serialize(self, graph_module: torch.fx.GraphModule) -> GraphModule: |
| 1358 | graph = self.serialize_graph(graph_module) |
| 1359 | |
| 1360 | return GraphModule( |
| 1361 | graph=graph, |
| 1362 | signature=self.serialize_signature(self.graph_signature), |
| 1363 | module_call_graph=self.serialize_module_call_graph(self.module_call_graph), |
| 1364 | ) |
| 1365 | |
| 1366 | |
| 1367 | class ExportedProgramSerializer: |
no test coverage detected