[WIP] reenable bfloat16 masked_select (#36859)
Summary:
Try reenabling bfloat16 masked_select, see it windows tests pass.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36859
Differential Revision: D21109535
Pulled By: ngimel
fbshipit-source-id: ca260943e6575d8e788e9fd87161a0d40d3d44fb
