Method
__init__
(
self,
graph_signature: ep.ExportGraphSignature,
module_call_graph: List[ep.ModuleCallEntry],
)
Source from the content-addressed store, hash-verified
| 434 | |
| 435 | class GraphModuleSerializer: |
| 436 | def __init__( |
| 437 | self, |
| 438 | graph_signature: ep.ExportGraphSignature, |
| 439 | module_call_graph: List[ep.ModuleCallEntry], |
| 440 | ): |
| 441 | self.graph_state = GraphState() |
| 442 | self.graph_signature = graph_signature |
| 443 | self.module_call_graph = module_call_graph |
| 444 | self.custom_objs: Dict[str, torch._C.ScriptObject] = {} |
| 445 | |
| 446 | @contextmanager |
| 447 | def save_graph_state(self): |
Callers
nothing calls this directly
Tested by
no test coverage detected