pytorch

0d0ebcdf - feature: adding the ability to restore shapes after loading a traced model (#90744)

feature: adding the ability to restore shapes after loading a traced model (#90744) Adds the ability to store inputs used in tracing models when calling torch.jit.save and restore the input shapes using torch.jit.load if the appropriate variables are set. Fixes [89185](https://github.com/pytorch/pytorch/issues/89185) Pull Request resolved: https://github.com/pytorch/pytorch/pull/90744 Approved by: https://github.com/davidberard98