Enable torch.where() to support Float16 & BFloat16 type inputs (#49004)
Summary:
Fixed https://github.com/pytorch/pytorch/issues/49075
Pull Request resolved: https://github.com/pytorch/pytorch/pull/49004
Reviewed By: zou3519
Differential Revision: D25495225
Pulled By: H-Huang
fbshipit-source-id: 09418ee5503f65c8862e40119c5802779505a4db