Infer types (#56832)
Summary:
Addresses: Infer argument types for functions in JIT
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56832
Reviewed By: pbelevich
Differential Revision: D27979495
Pulled By: nikithamalgifb
fbshipit-source-id: 82156a516c7f96cdd3f7a067d41cb210a6d13a51