pytorch/torch/_dynamo/variables
2024-04-13 02:07:35 +00:00
..
__init__.py [dynamo] Improve constant-prop for regex/torch.__version__ (#123705) 2024-04-12 19:03:13 +00:00
base.py [dynamo] Support __next__ on UserDefinedObjectVariable (#122565) 2024-03-31 19:00:03 +00:00
builder.py [dynamo] Graph break on uninitialized nn.Module (#123790) 2024-04-12 19:03:13 +00:00
builtin.py [dynamo][dict] Add UnspecializedNNModuleVariable to dict keys (#122812) 2024-04-13 02:07:35 +00:00
constant.py [dynamo] Remove VariableTracker.parents_tracker (#122058) 2024-03-19 04:23:24 +00:00
ctx_manager.py [dynamo] Support warnings.catch_warnings (#123511) 2024-04-08 22:27:46 +00:00
dicts.py [dynamo][dict] Add UnspecializedNNModuleVariable to dict keys (#122812) 2024-04-13 02:07:35 +00:00
distributed.py Don't create world pg variable out of thin air when rewriting c10d collectives (#122561) 2024-03-26 20:12:08 +00:00
functions.py [dynamo] Support custom __setattr__ on UserDefinedObjectVariable (#123318) 2024-04-07 21:06:52 +00:00
higher_order_ops.py [dynamo] Support Tuple[int] args to autograd.Function (#123887) 2024-04-12 19:03:13 +00:00
iter.py [dynamo] Improve constant-prop for regex/torch.__version__ (#123705) 2024-04-12 19:03:13 +00:00
lazy.py [dynamo] Replace VariableTracker.apply with visit/realize_all (#122218) 2024-03-20 07:53:18 +00:00
lists.py [dynamo][dict] Add UnspecializedNNModuleVariable to dict keys (#122812) 2024-04-13 02:07:35 +00:00
misc.py [dynamo] Support numpy.iinfo/finfo (#123803) 2024-04-12 19:03:13 +00:00
nn_module.py [dynamo][dict] Add UnspecializedNNModuleVariable to dict keys (#122812) 2024-04-13 02:07:35 +00:00
optimizer.py Remove special handling of step with closure (#123620) 2024-04-09 21:15:24 +00:00
sdpa.py [dynamo] Refactor reconstruct() not to return anything (#120150) 2024-02-17 17:13:41 +00:00
tensor.py [dynamo] Emit warning to turn on capture_scalar_outputs (#123896) 2024-04-12 19:03:13 +00:00
torch_function.py [dynamo] Optimize SourcelessBuilder (#122063) 2024-03-19 04:23:30 +00:00
torch.py Fix guard_size_oblivious on non-symbolic expression (#123743) 2024-04-10 22:45:54 +00:00
user_defined.py [dynamo] Fix @property on user-defined nn.Module (#123804) 2024-04-12 19:03:13 +00:00