Add BFloat16 support for isinf and isfinite (#49356)
Summary:
Also fix some tests.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/49356
Reviewed By: mruberry
Differential Revision: D25604364
Pulled By: ngimel
fbshipit-source-id: 9efdd83aaa96cacc66e9689db9f9d8c24175a693