pytorch
faf0a3bd - Move bernoulli_() to DistributionTemplates (#38558)

Commit

4 years ago

Move bernoulli_() to DistributionTemplates (#38558) Summary: resolve the feature introduced in https://github.com/pytorch/pytorch/issues/37373 Pull Request resolved: https://github.com/pytorch/pytorch/pull/38558 Differential Revision: D21920685 Pulled By: pbelevich fbshipit-source-id: 50c77d9aaa334b3276a2352afe6c4ad03f12be31

Author

Kiyosora

Committer

facebook-github-bot

Parents

a25b1b91

Files11

aten/src/ATen
- native
  - DistributionTemplates.h
  - Distributions.cpp
  - UnaryOps.h
  - cpu
    - DistributionTemplates.h
    - UnaryOpsKernel.cpp
  - cuda
    - DistributionBernoulli.cu
    - DistributionTemplates.h
  - native_functions.yaml
- test
  - cpu_rng_test.cpp
test
- test_cuda.py
- test_torch.py

pytorch faf0a3bd - Move bernoulli_() to DistributionTemplates (#38558)

pytorch
faf0a3bd - Move bernoulli_() to DistributionTemplates (#38558)