pytorch/torch/_functorch
2024-11-02 00:10:17 +00:00
..
_aot_autograd Update tensorify pass to specialize symfloats we didn't tensorify away (#138868) 2024-11-01 03:18:02 +00:00
__init__.py
aot_autograd.py Add option to dynamo_timed and chromium_event_logger for logging pt2 compile events (#139309) 2024-11-01 02:40:25 +00:00
apis.py Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-08-07 00:05:20 +00:00
autograd_function.py Fix torch.library.register_vmap (#137306) 2024-10-04 03:46:35 +00:00
batch_norm_replacement.py Revert "[BE] typing for decorators - _library/custom_ops (#131578)" 2024-07-28 03:29:32 +00:00
benchmark_utils.py [Functorch] Fix devices Parameter Type in benchmark_utilization Function (#138774) 2024-10-25 19:25:18 +00:00
compile_utils.py Fix unintentional deduplication of returned tensors (#134726) 2024-09-04 23:42:56 +00:00
compilers.py [BE] UFMT directory torch/_functorch (#123723) 2024-04-12 08:04:51 +00:00
config.py Class rename (#139490) 2024-11-02 00:10:17 +00:00
deprecated.py Flip default value for mypy disallow_untyped_defs [2/11] (#127839) 2024-06-08 18:23:08 +00:00
eager_transforms.py Fix linearize(grad(...)) call (#133364) 2024-08-15 17:55:36 +00:00
functional_call.py Allow optional positional arguments for torch.func.functional_call (#134643) 2024-09-12 15:22:06 +00:00
fx_minifier.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
make_functional.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
partitioners.py Refactor FxGraphDrawer to use HTML-like labels (#137726) 2024-11-01 23:19:50 +00:00
pyfunctorch.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
python_key.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
pytree_hacks.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
top_operators_github_usage.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
utils.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
vmap.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00