pytorch/torch/_decomp
Chien-Lin Chen 161425ff9f Added aten.bernoulli.p and aten.bernoulli.default decompositions (#139141)
Fixes #105519

Added aten.bernoulli.p decomposition and moved/rewrote aten.bernoulli.deafult to make them included in core aten decomposition.

Tested the sample code in [105519](https://github.com/pytorch/pytorch/issues/105519), torch.bernoulli could be decomposed by the code snippet.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/139141
Approved by: https://github.com/eellison
2024-11-20 19:52:57 +00:00
..
__init__.py Added aten.bernoulli.p and aten.bernoulli.default decompositions (#139141) 2024-11-20 19:52:57 +00:00
decompositions.py Added aten.bernoulli.p and aten.bernoulli.default decompositions (#139141) 2024-11-20 19:52:57 +00:00
decompositions_for_jvp.py
decompositions_for_rng.py