pytorch
51e341df - [bernoulli_kernel] Replace CPU_tensor_apply functions with cpu_serial_kernel (#39711)

Commit
4 years ago
[bernoulli_kernel] Replace CPU_tensor_apply functions with cpu_serial_kernel (#39711) Summary: Resolve https://github.com/pytorch/pytorch/issues/39556 Related https://github.com/pytorch/pytorch/issues/38558 Replace CPU_tensor_apply functions with cpu_serial_kernel in bernoulli_kernel, unifying bernoulli_kernel with all other kernels in `cpu/DistributionTemplates.h`. Signed-off-by: Xiong Wei <xiongw.fnst@cn.fujitsu.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/39711 Differential Revision: D22052374 Pulled By: pbelevich fbshipit-source-id: 416334da50195b67f05a18a98971f370cba4fb0d
Parents
Loading