Bucket of shape analysis fixes (#41044)
* [JIT] fix unfold shape analysis (#40749)
Summary:
unfold on a 0-dimensioned tensor returns a 1-dim tensor
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40749
Differential Revision: D22361481
Pulled By: eellison
fbshipit-source-id: 621597e5f97f6e39953eb86f8b85bb4142527a9f
* shape analysis fix for default dtype'
ghstack-source-id: 723aa27c2685417715a0891f5ca1ae885d4c9832
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40938
* fix grad thrashing of shape analysis
ghstack-source-id: dd8742b1da52d17e9d6ab6c81ff0b27520b09417
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40939
Co-authored-by: Elias Ellison <eellison@fb.com>