pytorch/torch/_subclasses
2023-12-20 21:36:25 +00:00
..
__init__.py
fake_tensor.py Add support for torch.cond in vmap (#114523) 2023-12-20 19:54:38 +00:00
fake_utils.py Expose Flash attn to autograd (#114378) 2023-12-01 23:42:06 +00:00
functional_tensor.py pre_dispatch aot_export (#115188) 2023-12-20 21:36:25 +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