pytorch/torch/_functorch
2023-07-26 08:07:31 +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 Revert "Slightly improve AOTAutograd logging with ViewAndMutationMeta (#105702)" 2023-07-25 20:17:27 +00:00
apis.py [pt2] grad support (#102264) 2023-06-21 10:13:09 +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 inductor: fix CSE issue when have symbolic shape input at the freezing path (#105651) 2023-07-26 08:07:31 +00:00
compilers.py Fix flake8 lint errors - part 2 - manual fixes (#99799) 2023-04-24 06:03:26 +00:00
config.py Revert "Re-enable low memory dropout (#103330)" 2023-07-05 19:00:40 +00:00
deprecated.py [pt2] grad support (#102264) 2023-06-21 10:13:09 +00:00
eager_transforms.py [BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432) 2023-07-19 13:48:44 +00:00
functional_call.py [Reland] Update mypy to 1.4.1 (#105227) 2023-07-15 20:30:20 +00:00
fx_minifier.py Add a little more error checking to minifier (#103057) 2023-06-07 14:40:12 +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 [partitioners][ac][dynamic] Fix output signature of fwd with symints (#105771) 2023-07-22 03:04:11 +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 [BE] Enable ruff's UP rules and autoformat dynamo / functorch and refs (#105432) 2023-07-19 13:48:44 +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