pytorch/torch/_dynamo/variables
2023-06-28 13:13:07 +00:00
..
__init__.py [dynamo][numpy] Add support for builtin functions (#103457) 2023-06-15 09:18:45 +00:00
base.py Lift user defined attributes into inputs for certain cases (user defined types and tensors) (#103386) 2023-06-20 23:45:19 +00:00
builder.py Specialize storage_offset - Does not cover automatic dynamic (#104204) 2023-06-27 05:51:42 +00:00
builtin.py fix specialization when you pass an unspec int into slicing on a Python list. (#104142) 2023-06-28 13:13:07 +00:00
constant.py [BE] Enable C419 rule for any all shortcircuiting (#99890) 2023-04-25 15:02:13 +00:00
ctx_manager.py [Dynamo] Fix torch.{cuda/cpu}.amp.autocast arguments binding bug (#101052) 2023-05-10 21:19:18 +00:00
dicts.py Simulate treespec flattening/unflattening (#101896) 2023-06-23 10:53:15 +00:00
functions.py Dynamo remaps legacy allgather to traceable one (#102232) 2023-05-30 16:45:25 +00:00
lists.py fix specialization when you pass an unspec int into slicing on a Python list. (#104142) 2023-06-28 13:13:07 +00:00
misc.py Fix _saved_tensors argument issue in test (#103594) 2023-06-20 19:03:41 +00:00
nn_module.py [Dynamo] No graph break for super(MyConv{1/2/3}d, self).forward and super(MyConvTranspose, self).forward (#102509) 2023-06-06 22:01:17 +00:00
optimizer.py Fix optimizer grad mode state interaction with dynamo (#103952) 2023-06-23 02:07:08 +00:00
tensor.py fix specialization when you pass an unspec int into slicing on a Python list. (#104142) 2023-06-28 13:13:07 +00:00
torch.py [dynamo][higher order op] Relaxing too restrictive check for output to be a list/tuple of tensors (#104221) 2023-06-28 00:30:43 +00:00
user_defined.py [dynamo] If UserDefinedObjectVariable.var_getattr() is a callable, try handling as a TorchVariable (#104231) 2023-06-28 02:39:03 +00:00