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