bfloat16: enable basic math function (#35172)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/35172
Test Plan: Imported from OSS
Differential Revision: D20721068
Pulled By: ngimel
fbshipit-source-id: 7e40bda6683f041f04f78739a950cb2a6ac74571