add `OpInfo` for `torch.nn.functional.one_hot` (#62253)
Summary:
Addresses facebookresearch/functorch#78.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62253
Reviewed By: heitorschueroff
Differential Revision: D29992924
Pulled By: zou3519
fbshipit-source-id: 1fc81edf3c8ca0722c5db0b32929a4cb3285f634