pytorch/torch/_decomp
2022-09-28 23:06:59 +00:00
..
__init__.py dispatch API for checking computed table, use it in prim decomps (#82358) 2022-08-10 23:42: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 Turn on aliasing tests for fake backwards, Fix Batch norm running mean/var decomp aliasing (#85471) 2022-09-28 23:06:59 +00:00