pytorch/torch/_dynamo/variables
2024-11-22 22:42:46 +00:00
..
__init__.py [dynamo] Represent all cells as NewCellVariable (#140153) 2024-11-15 17:17:30 +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 jk for unspecialize float killswitch (#141143) 2024-11-20 23:20:52 +00:00
builtin.py [dynamo] fix sorted(...) when key function is explicitly passed with key=None (#141224) 2024-11-22 22:42:46 +00:00
constant.py [dynamo] Simplify ConstantVariable.create and ConstantVariable.__init__ (#140745) 2024-11-19 00:22:50 +00:00
ctx_manager.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +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 [dynamo] Remove closure_cells and merge/remove code paths (#140154) 2024-11-15 17:17:30 +00:00
higher_order_ops.py [while_loop] support pytree inputs (#140059) 2024-11-20 21:12:29 +00:00
iter.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
lazy.py [dynamo] Replace __str__ with __repr__ in some places (#136316) 2024-10-21 19:50:38 +00:00
lists.py [dyanmo] support subclasses of namedtuple type (#140534) 2024-11-17 14:13:40 +00:00
misc.py [dynamo] Represent all cells as NewCellVariable (#140153) 2024-11-15 17:17:30 +00:00
nn_module.py [user-empathy-day][dynamo] Remove speical casing for torch.nn.Parameter tracing (#141209) 2024-11-21 21:18:42 +00:00
optimizer.py Ensure index for state guard construction is a source (#140515) 2024-11-15 22:02:50 +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 [dynamo] Support is comparison for symnodes (#140754) 2024-11-19 00:19:33 +00:00
torch_function.py [dynamo] Simplify creation of VariableTrackers (#135714) 2024-10-18 09:36:46 +00:00
torch.py [dynamo] Restrict support for out= variants of torch operators (#140202) 2024-11-13 03:14:23 +00:00
user_defined.py [user empathy day][dynamo] Support get on subclassed dicts (#141214) 2024-11-21 21:18:42 +00:00