| .. |
|
_aot_autograd
|
Consider storage_changed for assigning alias_of_input in aot_autograd when computing differentiable outputs that alias each other (#115315)
|
2023-12-12 23:21:58 +00:00 |
|
__init__.py
|
Revert "Replace _dynamo.config with an object instead of module (#96455)"
|
2023-06-05 07:06:14 +00:00 |
|
aot_autograd.py
|
Fix https://github.com/pytorch/pytorch/issues/114892 (#115054)
|
2023-12-04 18:29:33 +00:00 |
|
apis.py
|
[pt2] support vmap (#101707)
|
2023-08-09 03:39:33 +00:00 |
|
autograd_function.py
|
Use pytree.arg_tree_leaves everywhere (#112394)
|
2023-10-31 15:57:06 +00:00 |
|
batch_norm_replacement.py
|
[functorch] move batch_norm_replacement to torch.func (#91412)
|
2023-01-12 19:15:41 +00:00 |
|
benchmark_utils.py
|
Fix typo under torch/_functorch directory (#111067)
|
2023-10-11 23:09:36 +00:00 |
|
compile_utils.py
|
Use pytree.tree_leaves everywhere (#112324)
|
2023-10-30 03:39:04 +00:00 |
|
compilers.py
|
Prefer e.is_number over not e.free_symbols in SymPy (#112688)
|
2023-11-06 20:05:13 +00:00 |
|
config.py
|
[utils] move config_typing.pyi to torch.utils (#113929)
|
2023-11-17 18:51:57 +00:00 |
|
deprecated.py
|
[pt2] support vmap (#101707)
|
2023-08-09 03:39:33 +00:00 |
|
eager_transforms.py
|
Use pytree.arg_tree_leaves everywhere (#112394)
|
2023-10-31 15:57:06 +00:00 |
|
functional_call.py
|
Fix typo under torch/_functorch directory (#111067)
|
2023-10-11 23:09:36 +00:00 |
|
fx_minifier.py
|
Add function to port FX minified graph to HLO via StableHLO (#109084)
|
2023-10-02 19:36:04 +00:00 |
|
make_functional.py
|
[BE] [2/3] Rewrite super() calls in functorch and torch (#94588)
|
2023-02-10 21:16:33 +00:00 |
|
partitioners.py
|
[inductor] add a config to specify the shape attribute for the generated svg graphs (#114811)
|
2023-11-30 06:10:37 +00:00 |
|
pyfunctorch.py
|
[BE]: Bugfix functorch and some generic typing improvements (#101337)
|
2023-05-14 14:20:56 +00:00 |
|
python_key.py
|
[Reland] Move functorch/_src to torch/_functorch (#88756) (#90091)
|
2022-12-03 14:17:15 +00:00 |
|
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
|
[BE]: Update ruff to v0.0.290 (#109435)
|
2023-09-16 18:43:34 +00:00 |
|
utils.py
|
[pt2] support vmap (#101707)
|
2023-08-09 03:39:33 +00:00 |
|
vmap.py
|
[functorch] fix potential race condition while loading vmap decomposition library (#113520)
|
2023-11-20 19:50:54 +00:00 |
76ced0df03