pytorch/torch/_dynamo/variables
2024-05-29 16:00:15 +00:00
..
__init__.py [dynamo] Improve constant-prop for regex/torch.__version__ (#123705) 2024-04-12 19:03:13 +00:00
base.py Add VariableTracker.debug_repr (#126299) 2024-05-15 23:55:29 +00:00
builder.py [dynamo] Handle np.iinfo/finfo/dtype as input (#124482) 2024-05-29 16:00:15 +00:00
builtin.py [dynamo] make callable(nn_module) return True (#127026) 2024-05-24 18:31:43 +00:00
constant.py [dynamo] Support enum construction (#127364) 2024-05-29 08:09:21 +00:00
ctx_manager.py [dynamo] fix crash when context manager is passed to a function (#125321) 2024-05-03 23:01:30 +00:00
dicts.py [Dynamo] Support SET_UPDATE (#126243) 2024-05-16 20:05:34 +00:00
distributed.py support as_python_constant on PlacementClassVariable (#124398) 2024-05-01 21:56:01 +00:00
functions.py [dynamo] trace through nn parametrize (#125771) 2024-05-09 17:43:48 +00:00
higher_order_ops.py [cond] support torch built in function as subgraph (#126909) 2024-05-24 18:31:43 +00:00
iter.py [dynamo] Improve constant-prop for regex/torch.__version__ (#123705) 2024-04-12 19:03:13 +00:00
lazy.py [dynamo] fix crash when context manager is passed to a function (#125321) 2024-05-03 23:01:30 +00:00
lists.py Add VariableTracker.debug_repr (#126299) 2024-05-15 23:55:29 +00:00
misc.py [dynamo] Support numpy.dtype (#124481) 2024-05-29 14:45:14 +00:00
nn_module.py [dynamo] Detect monkeypatching on nn module forward method (#126203) 2024-05-15 20:41:13 +00:00
optimizer.py Fix capturable enablement conditions (#125826) 2024-05-11 06:29:59 +00:00
script_object.py [dynamo] support torchbind object input (#124978) 2024-05-07 03:02:00 +00:00
sdpa.py Remove removed ruff rule TRY200 (#126256) 2024-05-17 16:31:05 +00:00
tensor.py [dynamo][TensorVariable] Support "if param.grad_fn" usecase (#126960) 2024-05-25 01:09:26 +00:00
torch_function.py [BE]: Try TCH autofixes on torch/ (#125536) 2024-05-05 23:13:59 +00:00
torch.py generalize custom_fwd&custom_bwd to be device-agnostic (#126531) 2024-05-25 06:48:16 +00:00
user_defined.py [dynamo] Support enum construction (#127364) 2024-05-29 08:09:21 +00:00