Jason Ansel 3c8c7e2a46 [dynamo] Tweak naming for module hook bw_state (#121609) ... Some minor changes not related to the other PRs in the stack Pull Request resolved: https://github.com/pytorch/pytorch/pull/121609 Approved by: https://github.com/yanboliang		2024-03-12 16:27:56 +00:00
..
_aot_autograd	[dynamo] Tweak naming for module hook bw_state (#121609)	2024-03-12 16:27:56 +00:00
__init__.py
aot_autograd.py	[hoo] Add with_effects to handle side effectful ops (#120296)	2024-03-05 08:58:32 +00:00
apis.py	Let torch dynamo inline torch.func.grad (#118407)	2024-02-28 20:05:00 +00:00
autograd_function.py	Enable possibly-undefined error code (#118533)	2024-01-30 21:07:01 +00:00
batch_norm_replacement.py
benchmark_utils.py	Replace follow_imports = silent with normal (#118414)	2024-01-27 02:44:11 +00:00
compile_utils.py	Fixed hash issue in fx_graph_cse (#119567)	2024-02-12 18:52:11 +00:00
compilers.py	Replace follow_imports = silent with normal (#118414)	2024-01-27 02:44:11 +00:00
config.py	Expose aggressive_recomputation as an inductor config (#118943)	2024-02-03 00:17:03 +00:00
deprecated.py	Let torch dynamo inline torch.func.grad (#118407)	2024-02-28 20:05:00 +00:00
eager_transforms.py	Teach dynamo about vjp (#119405)	2024-03-01 00:21:10 +00:00
functional_call.py	Fix typo under torch/_functorch directory (#111067)	2023-10-11 23:09:36 +00:00
fx_minifier.py	Replace follow_imports = silent with normal (#118414)	2024-01-27 02:44:11 +00:00
make_functional.py
partitioners.py	Revert "Batch Norm Consolidation (#116092)"	2024-03-11 22:22:41 +00:00
pyfunctorch.py	Update vmap guard to avoid recompilations (#119061)	2024-02-13 20:50:23 +00:00
python_key.py
pytree_hacks.py	[pytree][BE] reorganize imports and format code style and update type hints (#112268)	2023-10-28 16:30:24 +00:00
top_operators_github_usage.py	Replace follow_imports = silent with normal (#118414)	2024-01-27 02:44:11 +00:00
utils.py	[pt2] support vmap (#101707)	2023-08-09 03:39:33 +00:00
vmap.py	Replace follow_imports = silent with normal (#118414)	2024-01-27 02:44:11 +00:00