pytorch/torch/_dynamo/variables
2024-11-26 01:10:08 +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] match implementation for sorted(...) with CPython (#141227) 2024-11-24 20:01:50 +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] Fix and simplify hanlding of Set.update method (#141286) 2024-11-26 00:41:50 +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 [Dynamo][autograd.Function] Use fake tensor prop to infer fwd output (#136184) 2024-11-26 01:10:08 +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 [dynamo] support list.sort sort non-constant iterable with constant keys (#141485) 2024-11-25 21:06:11 +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