Replace THPLayout with at::Layout in Python Argument Parser (#34543) (#34584)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34584
Test Plan:
```
python setup.py develop
python test/test_torch.py
```
Output:
```
...
Ran 3834 tests in 198.825s
OK (skipped=180)
```
Imported from OSS
Differential Revision: D20403330
fbshipit-source-id: 41474d5e7001db070f98ac8379f909f0ac74deb6