pytorch/torch/_dynamo/variables
2024-02-13 20:50:23 +00:00
..
__init__.py [dynamo] Support torch.distributed.fsdp._flat_param._same_storage_size (#119627) 2024-02-13 01:27:37 +00:00
base.py [dynamo] Improve support for backwards hooks (#119525) 2024-02-10 01:14:03 +00:00
builder.py [Dynamo] Minor cleanup of redundant function lookup logics (#119666) 2024-02-12 19:00:39 +00:00
builtin.py [dynamo] inlining into __iter__ of user defined object (#119243) 2024-02-08 17:07:30 +00:00
constant.py Print the value of constants in __str__ (#119276) 2024-02-08 16:23:36 +00:00
ctx_manager.py Update vmap guard to avoid recompilations (#119061) 2024-02-13 20:50:23 +00:00
dicts.py [Dynamo][17/N] Rename SkipFilesVariable to SkipFunctionVariable and move to functions.py (#119619) 2024-02-10 19:33:37 +00:00
distributed.py Unify MYPYINDUCTOR and MYPY (#118432) 2024-01-27 17:23:20 +00:00
functions.py Support FunctoolsPartialVariable::get_function, fix NamedTupleVariable::as_proxy and handle call_function in get_fake_values_from_nodes (#119435) 2024-02-13 01:44:08 +00:00
higher_order_ops.py [export] Convert internal tests to using .module() (#119105) 2024-02-08 17:19:07 +00:00
iter.py Unify MYPYINDUCTOR and MYPY (#118432) 2024-01-27 17:23:20 +00:00
lazy.py Unify MYPYINDUCTOR and MYPY (#118432) 2024-01-27 17:23:20 +00:00
lists.py Support FunctoolsPartialVariable::get_function, fix NamedTupleVariable::as_proxy and handle call_function in get_fake_values_from_nodes (#119435) 2024-02-13 01:44:08 +00:00
misc.py [Dynamo][17/N] Rename SkipFilesVariable to SkipFunctionVariable and move to functions.py (#119619) 2024-02-10 19:33:37 +00:00
nn_module.py [Dynamo][16/N] Move skipfiles to trace_rules.py (#119432) 2024-02-09 18:18:23 +00:00
optimizer.py [dynamo,optim] Use the actual sources from the parameters when tracing "params" in an optimizer (#118535) 2024-02-02 14:42:56 +00:00
sdpa.py Unify MYPYINDUCTOR and MYPY (#118432) 2024-01-27 17:23:20 +00:00
tensor.py [dynamo] Capture untyped_storage().resize_() (#119647) 2024-02-13 19:03:28 +00:00
torch_function.py Unify MYPYINDUCTOR and MYPY (#118432) 2024-01-27 17:23:20 +00:00
torch.py add get_function to TorchInGraphFunctionVariable (#119314) 2024-02-12 16:35:34 +00:00
user_defined.py [Dynamo] Minor cleanup of redundant function lookup logics (#119666) 2024-02-12 19:00:39 +00:00