| .. |
|
_activation_checkpointing
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
_aot_autograd
|
Revert "[compiled autograd] Always proxy autograd.Function nodes; handle AOT backwards (#143405)"
|
2025-01-23 23:34:13 +00:00 |
|
__init__.py
|
|
|
|
aot_autograd.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
apis.py
|
[BE] typing for decorators (#144161)
|
2025-01-04 16:40:09 +00:00 |
|
autograd_function.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
batch_norm_replacement.py
|
remove allow-untyped-defs for torch/_functorch/batch_norm_replacement.py (#143438)
|
2024-12-18 09:01:06 +00:00 |
|
benchmark_utils.py
|
|
|
|
compile_utils.py
|
|
|
|
compilers.py
|
Remove unused Python variables in torch/[_-a]* (#133492)
|
2024-12-12 17:39:14 +00:00 |
|
config.py
|
[reland][export] don't decompose custom triton op when exporting (#144284)
|
2025-01-11 01:34:35 +00:00 |
|
deprecated.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
eager_transforms.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
functional_call.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
fx_minifier.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
make_functional.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
partitioners.py
|
[BE]: Simplify set add with set update (#145152)
|
2025-01-23 20:18:13 +00:00 |
|
pyfunctorch.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
python_key.py
|
|
|
|
pytree_hacks.py
|
|
|
|
top_operators_github_usage.py
|
|
|
|
utils.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |
|
vmap.py
|
PEP585 update - torch/_functorch (#145139)
|
2025-01-19 07:06:10 +00:00 |