pytorch/torch/_dynamo/variables
2024-03-04 14:38:45 +00:00
..
__init__.py [dynamo] Support module backwards hooks (#120685) 2024-03-01 02:24:26 +00:00
base.py [dynamo] Improve support for backwards hooks (#119525) 2024-02-10 01:14:03 +00:00
builder.py Add sparse compressed fake tensor support (#120920) 2024-03-04 14:38:45 +00:00
builtin.py [dynamo][comp-time] BuiltinVariableTracker - inspect signature only on failure (#121053) 2024-03-02 23:03:00 +00:00
constant.py Print the value of constants in __str__ (#119276) 2024-02-08 16:23:36 +00:00
ctx_manager.py [dynamo] Fix inference_mode context variable (#120830) 2024-02-29 17:10:06 +00:00
dicts.py [dynamo] Refactor reconstruct() not to return anything (#120150) 2024-02-17 17:13:41 +00:00
distributed.py [dynamo] Support module backwards hooks (#120685) 2024-03-01 02:24:26 +00:00
functions.py Ban reset_to_zero argument to triton.autotune in user defined kernels (#120938) 2024-03-01 02:37:24 +00:00
higher_order_ops.py [Compiled Autograd] Introduce BackwardState capture (#120382) 2024-02-28 20:36:47 +00:00
iter.py Unify MYPYINDUCTOR and MYPY (#118432) 2024-01-27 17:23:20 +00:00
lazy.py Unify MYPYINDUCTOR and MYPY (#118432) 2024-01-27 17:23:20 +00:00
lists.py [dynamo] Refactor reconstruct() not to return anything (#120150) 2024-02-17 17:13:41 +00:00
misc.py [dynamo] Reorder logs (#116106) 2024-03-01 17:04:24 +00:00
nn_module.py [Dynamo] Support lazy module with namedtuple/dict input (#119972) 2024-02-15 23:18:18 +00:00
optimizer.py [dynamo,optim] Use the actual sources from the parameters when tracing "params" in an optimizer (#118535) 2024-02-02 14:42:56 +00:00
sdpa.py [dynamo] Refactor reconstruct() not to return anything (#120150) 2024-02-17 17:13:41 +00:00
tensor.py Add sparse compressed fake tensor support (#120920) 2024-03-04 14:38:45 +00:00
torch_function.py Fix handling none source in build_torch_function_fn (#119724) 2024-02-15 21:21:47 +00:00
torch.py [dynamo] Minor refactors (#120966) 2024-03-03 02:20:48 +00:00
user_defined.py [dynamo] Minor refactors (#120966) 2024-03-03 02:20:48 +00:00