[jit] Shape propagation batch_norm, dropout, quantize, hardswidh (#64740)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64740
Test Plan: Imported from OSS
Reviewed By: eellison
Differential Revision: D30839111
Pulled By: IvanKobzarev
fbshipit-source-id: c8f477ee05769865c0a23127b7f8a8276f46b54e
