pytorch/torch/_dynamo/variables
2024-07-30 13:11:24 +00:00
..
__init__.py Revert "Let dynamo inline functional_call (#128646)" 2024-07-29 16:26:50 +00:00
base.py Remove mypy ignore from torch/_dynamo/variables/lazy.py (#131785) 2024-07-28 17:13:53 +00:00
builder.py [dynamo] lazily guard and specialize on the symint when used in f-string. (#131529) 2024-07-25 16:16:34 +00:00
builtin.py [dynamo] Support dict unpack of MutableMapping objects (#131961) 2024-07-30 05:49:58 +00:00
constant.py [dynamo] add lazy IteratorVariable implementations for map and zip (#131413) 2024-07-26 10:47:38 +00:00
ctx_manager.py Relax constraints for creating a GenericContextWrappingVariable (#129091) 2024-07-29 15:40:59 +00:00
dicts.py [dynamo] Support dict unpack of MutableMapping objects (#131961) 2024-07-30 05:49:58 +00:00
distributed.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
functions.py [dynamo][log] Suggest to use pytree when graph-break on optree (#131827) 2024-07-30 05:49:58 +00:00
higher_order_ops.py Revert "Let dynamo inline functional_call (#128646)" 2024-07-29 16:26:50 +00:00
iter.py [dynamo] make more unpack_var_sequence calls forced (#132069) 2024-07-30 02:30:08 +00:00
lazy.py Remove mypy ignore from torch/_dynamo/variables/lazy.py (#131785) 2024-07-28 17:13:53 +00:00
lists.py [dynamo] add lazy IteratorVariable implementations for map and zip (#131413) 2024-07-26 10:47:38 +00:00
misc.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
nn_module.py [dynamo] Do not guard on keys for _forward_hooks and _forward_pre_hooks (#131682) 2024-07-26 04:39:54 +00:00
optimizer.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
script_object.py [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
sdpa.py Revert "Grouped Query Attention (#128898)" 2024-07-30 13:11:24 +00:00
tensor.py dynamo: mutations on .data should be invisible to autograd (#131403) 2024-07-26 14:22:20 +00:00
torch_function.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
torch.py Annotate all InstructionTranslator (#131509) 2024-07-24 23:45:53 +00:00
user_defined.py [dynamo] make more unpack_var_sequence calls forced (#132069) 2024-07-30 02:30:08 +00:00