pytorch/torch/_dynamo/variables
2024-11-11 23:38:00 +00:00
..
__init__.py [dynamo][weakref] Support weakref.ref call (#139914) 2024-11-06 23:16:41 +00:00
base.py [dynamo] Add is_mutable() and is_immutable() methods to VariableTracker (#139341) 2024-11-05 19:11:41 +00:00
builder.py add special case for __round__ constant variables (#139583) 2024-11-09 03:25:53 +00:00
builtin.py [Dynamo] Support tensor mro without source (#139838) 2024-11-06 08:52:53 +00:00
constant.py add special case for __round__ constant variables (#139583) 2024-11-09 03:25:53 +00:00
ctx_manager.py [Dynamo] Trace enter/exit of TorchFunctionModes (#135422) (#137114) 2024-10-09 02:29:40 +00:00
dicts.py [dynamo] Add is_mutable() and is_immutable() methods to VariableTracker (#139341) 2024-11-05 19:11:41 +00:00
distributed.py Revert "[dynamo] Fix constant propagation in builtins and UserClasses (#131354)" 2024-11-01 00:13:20 +00:00
functions.py Revert "[dynamo] Remove dead code path for capturing __class__ in UserFunctionVariable (#140034)" 2024-11-11 23:38:00 +00:00
higher_order_ops.py Add base class for single-subgraph inductor HOPs (#139898) 2024-11-11 16:12:35 +00:00
iter.py [dynamo] Add is_mutable() and is_immutable() methods to VariableTracker (#139341) 2024-11-05 19:11:41 +00:00
lazy.py [dynamo] Replace __str__ with __repr__ in some places (#136316) 2024-10-21 19:50:38 +00:00
lists.py [dynamo] Add is_mutable() and is_immutable() methods to VariableTracker (#139341) 2024-11-05 19:11:41 +00:00
misc.py [Dynamo] allow dynamic callables on tensor variables (#137940) 2024-11-08 23:49:34 +00:00
nn_module.py [dynamo][NFC] Rename mutable_local and add documentation (#139339) 2024-11-05 19:11:41 +00:00
optimizer.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
script_object.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
sdpa.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
tensor.py Make size a property with an assertion (#139794) 2024-11-09 03:39:41 +00:00
torch_function.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
torch.py Make size a property with an assertion (#139794) 2024-11-09 03:39:41 +00:00
user_defined.py [dynamo][user-defined] Walk __mro__ to get the member descriptor source (#140300) 2024-11-11 23:16:48 +00:00