pytorch/torch/_subclasses
Tugsbayasgalan Manlaibaatar fa1e89b337 Ban mutation on dropout outputs in export (#117879)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/117879
Approved by: https://github.com/ezyang
ghstack dependencies: #117811
2024-01-21 04:53:40 +00:00
..
__init__.py
fake_tensor.py Initial implementation of FakeTensor caching (#113873) 2024-01-17 20:38:54 +00:00
fake_utils.py Expose Flash attn to autograd (#114378) 2023-12-01 23:42:06 +00:00
functional_tensor.py Ban mutation on dropout outputs in export (#117879) 2024-01-21 04:53:40 +00:00
meta_utils.py Add markDynamoStrictTest to NT tests (#116111) 2023-12-20 20:13:20 +00:00
schema_check_mode.py Use pytree.tree_leaves everywhere (#112324) 2023-10-30 03:39:04 +00:00