[FX] Add torch.memory_format as a BaseArgumentType (#62593)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/62498
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62593
Reviewed By: H-Huang
Differential Revision: D30104091
Pulled By: cpuhrsch
fbshipit-source-id: 25b7a4b308219860c969db54d7b1867b1aa4180a
