Override shape hints with real weight shape extracted from workspace (#40872)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40872
Shape hints as its name suggests, is hint. We should use real shape from workspace for the weights.
Reviewed By: ChunliF
Differential Revision: D22337680
fbshipit-source-id: e7a6101fb613ccb332c3e34b1c2cb8c6c47ce79b