add BFloat16 support for kthvalue and median on CPU (#60074)
Summary:
Add BFloat16 support for kthvalue and median on CPU
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60074
Reviewed By: gchanan
Differential Revision: D29230348
Pulled By: heitorschueroff
fbshipit-source-id: fa9c086758d51069acf270faa526e4b141b0ef68