pytorch/torch/_dynamo/variables
2022-11-08 18:37:48 +00:00
..
__init__.py
base.py
builder.py [Dynamo] Support optimizing over any Tensor with requires_grad = True (#87141) 2022-10-19 22:13:07 +00:00
builtin.py [Dynamo] Improve BuiltinVariable log when incorrect arg count happens (#88409) 2022-11-05 00:17:18 +00:00
constant.py
dicts.py [Dynamo] Support optimizing over any Tensor with requires_grad = True (#87141) 2022-10-19 22:13:07 +00:00
functions.py [Dynamo] UserFunctionVariable supports type & ABCMeta as arguments (#88257) 2022-11-02 06:58:04 +00:00
lists.py Fix stupid N^2 naming behavior in FX and removed assert that slows things a lot sometimes (#87533) 2022-10-23 08:26:37 +00:00
misc.py [Dynamo] Fix bug: GradMode doesn't carry grad state correctly after graph break (#88537) 2022-11-07 18:03:31 +00:00
nn_module.py Revert "torchdynamo support modules() for nn_module (#88023)" 2022-11-08 18:37:48 +00:00
tensor.py Unify SymIntNode and SymFloatNode into SymNode (#87817) 2022-10-27 20:56:02 +00:00
torch.py [torchdynamo] support torch.autograd._profiler_enabled (#88378) 2022-11-07 20:36:26 +00:00
user_defined.py Revert "[EZ] Fix simple bug in torchdynamo (#87821)" 2022-10-28 06:11:42 +00:00