pytorch/torch/_functorch
2023-05-04 04:28:45 +00:00
..
__init__.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
aot_autograd.py Add AotAutogradFallbackTests to dynamic suite (#100454) 2023-05-04 04:28:45 +00:00
autograd_function.py [BE] Do not use unicode quotes (#99446) 2023-04-18 22:59:56 +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 Improve support for contextlib.nullcontext (#98111) 2023-04-02 02:33:14 +00:00
compile_utils.py [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
compilers.py Fix flake8 lint errors - part 2 - manual fixes (#99799) 2023-04-24 06:03:26 +00:00
config.py Delete use_functionalize feature flag (#99317) 2023-04-18 02:09:57 +00:00
deprecated.py Add deprecation messages for functorch.* function transforms (#92279) 2023-01-18 14:26:25 +00:00
eager_transforms.py [BE] Enable flake8-comprehension rule C417 (#97880) 2023-03-30 14:34:24 +00:00
functional_call.py documentation update (#98782) 2023-04-12 20:34:30 +00:00
fx_minifier.py Improve minifier printing to be more chatty when it makes sense (#100486) 2023-05-04 02:51:26 +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 Use a stable ordering for saved values in functorch.default_partition (#100111) 2023-05-02 05:14:31 +00:00
pyfunctorch.py Support functionalization on torch.cond (#89966) 2022-12-22 22:01:47 +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 [Reland] Move functorch/_src to torch/_functorch (#88756) (#90091) 2022-12-03 14:17:15 +00:00
top_operators_github_usage.py [BE] Remove unnecessary dict comprehensions (#97116) 2023-03-20 00:56:57 +00:00
utils.py Rename flag that enables/disables _SingleLevelFunction for functorch (#92025) 2023-01-17 13:36:41 +00:00
vmap.py Add vmap support for smooth_l1_loss_backward (#99429) 2023-04-28 10:58:07 +00:00