pytorch/torch/_dynamo/variables
2023-12-09 05:11:44 +00:00
..
__init__.py [Dynamo][6.1/N] Refactor out TorchInGraphFunctionVariable and improve heuristic (#113432) 2023-12-09 05:11:44 +00:00
base.py Refactor how HOPs create new args to subgraphs (#115185) 2023-12-06 01:29:03 +00:00
builder.py [Dynamo][6.1/N] Refactor out TorchInGraphFunctionVariable and improve heuristic (#113432) 2023-12-09 05:11:44 +00:00
builtin.py [Dynamo][6.1/N] Refactor out TorchInGraphFunctionVariable and improve heuristic (#113432) 2023-12-09 05:11:44 +00:00
constant.py [dynamo] Improve support for dynamic shapes str.format and _assert (#115203) 2023-12-06 04:54:45 +00:00
ctx_manager.py [contextlib] Wrapping a function with set_grad_enabled will consume its global mutation (#113359) 2023-11-09 19:16:20 +00:00
dicts.py [dynamo] Support hasattr on dataclass (#115046) 2023-12-05 19:01:14 +00:00
distributed.py [DeviceMesh] Rename _device_mesh.py to device_mesh.py to prepare for beta (#115099) (#115193) 2023-12-08 08:44:32 +00:00
functions.py [dynamo] Improve support for inspect.signature().parameters (#115047) 2023-12-04 19:08:36 +00:00
higher_order_ops.py Remove always restore (#115317) 2023-12-08 18:17:37 +00:00
iter.py [dynamo] Remove VariableTracker.add_options (#111725) 2023-11-07 19:55:19 +00:00
lazy.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
lists.py [dynamo] add list iterator contains (#115237) 2023-12-06 22:26:16 +00:00
misc.py Remove always restore (#115317) 2023-12-08 18:17:37 +00:00
nn_module.py [dynamo] Fix UnspecializedNNModuleVariable's source (#113852) 2023-11-17 08:17:27 +00:00
optimizer.py [dynamo] Eagerly install guards (#111415) 2023-11-07 19:55:19 +00:00
tensor.py Make subclass type instances constants (like UserDefinedClasses) (#115323) 2023-12-07 08:10:59 +00:00
torch_function.py [dynamo] Eagerly install guards (#111415) 2023-11-07 19:55:19 +00:00
torch.py [Dynamo][6.1/N] Refactor out TorchInGraphFunctionVariable and improve heuristic (#113432) 2023-12-09 05:11:44 +00:00
user_defined.py [dynamo] Add RestrictedListSubclassVariable (#115057) 2023-12-05 19:01:23 +00:00