jit trace (#59949)
Summary:
Fixes #{issue number}
Pull Request resolved: https://github.com/pytorch/pytorch/pull/59949
Reviewed By: ZolotukhinM
Differential Revision: D31366787
Pulled By: Krovatkin
fbshipit-source-id: 798cbcd97e8ecfba984f98cd70214954be9309af
