OpInfo: where (#58349)
Summary:
Reference: https://github.com/pytorch/pytorch/issues/54261
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58349
Reviewed By: mrshenli
Differential Revision: D28744220
Pulled By: mruberry
fbshipit-source-id: 893a2fb88a48a60df75c7d6e2f58a42ca949daa7
