pytorch/torch/_decomp
lezcano 154e58c032 Add most in-place references/decompositions (#88117)
We add most in-place references in a generic way. We also implement a
wrapper to implement the annoying interface that `nn.functional`
nonlinearities have.

We fix along the way a couple decompositions for some non-linearities by
extending the arguments that the references have.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88117
Approved by: https://github.com/mruberry
2022-11-18 14:59:46 +00:00
..
__init__.py Prefer python meta function over c++ meta function (#87426) 2022-10-25 16:49:02 +00:00
decompositions_for_jvp.py [reland 2] Call jit decomp in VariableType to improve forward AD coverage (#84976) 2022-09-15 22:46:19 +00:00
decompositions.py Add most in-place references/decompositions (#88117) 2022-11-18 14:59:46 +00:00