[ONNX] Fix assign input shape for tuple inputs & primitive type inputs (#54112) (#56164)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56164
Test Plan: Imported from OSS
Reviewed By: pbelevich
Differential Revision: D27866139
Pulled By: SplitInfinity
fbshipit-source-id: c59f5a07df685e1ccdc4860d603ec422ec80d188
