Elias Ellison d3f7c34cb3 Enable aten-aten decomps (#85921) ... Invokes aten-aten decomps with re-entrant FakeMode. These decomps are being used in other places, so it's good to unify the path static fake tensor takes / get additional testing etc. There is also an instance where we return different devices with cpu/cuda which this fixes ([batch_norm](https://github.com/pytorch/pytorch/blob/master/torch/_decomp/decompositions.py#L1374)) Pull Request resolved: https://github.com/pytorch/pytorch/pull/85921 Approved by: https://github.com/ezyang		2022-10-08 05:12:42 +00:00
..
__init__.py	Enable aten-aten decomps (#85921)	2022-10-08 05:12:42 +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	Enable aten-aten decomps (#85921)	2022-10-08 05:12:42 +00:00