Preserve graph signature throughout aimet_torch.export