pytorch/torch/_dynamo/variables
2024-01-18 09:34:50 +00:00
..
__init__.py [Dynamo][10/N] Remove TorchVariable and is_allowed (#116312) 2023-12-27 18:47:05 +00:00
base.py Remove replace_all and make VTs mutable (#113725) 2023-12-10 09:31:21 +00:00
builder.py [easy]Remove specialized value (#112252) 2024-01-18 09:34:50 +00:00
builtin.py [dynamo][easy]: Add support for operator.truth (#117463) 2024-01-14 19:08:31 +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 Reconstruct streams via global registration, temporary impl to unblock FSDP (#117386) 2024-01-13 07:03:33 +00:00
dicts.py Support NNModules as dict keys (#116723) 2024-01-09 03:32:47 +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 Impl. call_hasattr for BaseUserFunctionVariable (#116049) 2024-01-09 22:58:58 +00:00
higher_order_ops.py [Dynamo] Trace autograd.function in dynamo when inputs require grad (#116358) (#116897) 2024-01-16 03:57:13 +00:00
iter.py [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
lazy.py [dynamo] LazyVariable - redirect __str__ to the realized variable __str__ (#117583) 2024-01-17 21:12:12 +00:00
lists.py [dynamo] Add hasattr support for TupleVariable (#117694) 2024-01-18 07:47:43 +00:00
misc.py [Dynamo] Trace autograd.function in dynamo when inputs require grad (#116358) (#116897) 2024-01-16 03:57:13 +00:00
nn_module.py Support NNModules as dict keys (#116723) 2024-01-09 03:32:47 +00:00
optimizer.py Re-enable SGD (#117434) 2024-01-18 06:47:15 +00:00
tensor.py [easy]Remove specialized value (#112252) 2024-01-18 09:34:50 +00:00
torch_function.py Fix torch function kwarg dispatch (#117083) 2024-01-10 10:55:10 +00:00
torch.py [Dynamo] Make profiler function will be ignored warn only once (#117585) 2024-01-17 04:05:45 +00:00
user_defined.py Reland #111196 (take 4) "Support tensors as Dict keys" (#116934) 2024-01-07 01:37:26 +00:00