[jit] Set the graph input types before interpreting the graph during tracing (#68242)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68242
Test Plan: Imported from OSS
Reviewed By: saketh-are
Differential Revision: D32382958
Pulled By: navahgar
fbshipit-source-id: 4e82a604a9ea2046af2755de23944147e618a65f