pytorch/torch/_decomp
2023-06-21 20:57:43 +00:00
..
__init__.py [decomp] Add decomposition for torch.renorm (#103858) 2023-06-21 20:57:43 +00:00
decompositions.py [decomp] Don't call .item() in aten.fill.Tensor decomp (#103880) 2023-06-21 18:45:04 +00:00
decompositions_for_jvp.py
decompositions_for_rng.py [partitioner] fix for rng ops (#102123) 2023-05-25 00:35:07 +00:00