add BFloat16 support for bernoulli and Dropout on CPU (#56372)
authormingfeima <mingfei.ma@intel.com>
Wed, 25 Aug 2021 18:53:52 +0000 (11:53 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 25 Aug 2021 19:01:27 +0000 (12:01 -0700)
commitb0782f0f328321ab3ede798dfed3c7a143130e31
tree880710f858d7eaff836f6e917596645d56efd94a
parent72995657681f6173413b5ee7c62bd91212d07e8d
add BFloat16 support for bernoulli and Dropout on CPU (#56372)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56372

Test Plan: Imported from OSS

Reviewed By: heitorschueroff

Differential Revision: D28836792

Pulled By: VitalyFedyunin

fbshipit-source-id: ede951d172a59276e11383fd767778ab959b5a6b
aten/src/ATen/native/cpu/DistributionTemplates.h
aten/src/ATen/native/cpu/UnaryOpsKernel.cpp
test/test_nn.py
test/test_torch.py