pytorch/torch/_dynamo/variables
Jason Ansel c65c0682b1 [dynamo] Expand _nonvar_fields names (#111749)
This should be a small compile time optimization, since we won't need to
walk these fields in apply().

Pull Request resolved: https://github.com/pytorch/pytorch/pull/111749
Approved by: https://github.com/yanboliang
2023-10-23 02:58:16 +00:00
..
__init__.py [dynamo][stream]support device-agnostic stream in dynamo and capture stream/event method in fx graph (#108312) 2023-10-22 13:22:58 +00:00
base.py [dynamo] Expand _nonvar_fields names (#111749) 2023-10-23 02:58:16 +00:00
builder.py [dynamo][stream]support device-agnostic stream in dynamo and capture stream/event method in fx graph (#108312) 2023-10-22 13:22:58 +00:00
builtin.py [dynamo][stream]support device-agnostic stream in dynamo and capture stream/event method in fx graph (#108312) 2023-10-22 13:22:58 +00:00
constant.py Revert "[user errors] compulsory case names, allow multiple (#110878)" 2023-10-10 04:44:40 +00:00
ctx_manager.py [dynamo] Fix context wrapping grad mode variable (#111534) 2023-10-22 20:55:48 +00:00
dicts.py [dynamo] Add specialized variable tracker for sys.modules (#110990) 2023-10-13 20:08:40 +00:00
distributed.py [Easy] ConstantVariable() -> .create (#109896) 2023-09-22 22:30:15 +00:00
functions.py [dynamo] Forward fix a bunch of distributed collective allow fixes (#111171) 2023-10-13 15:49:04 +00:00
higher_order_ops.py [dynamo] share a subgraph tracer across fwd and bwd in autograd.Function (#111588) 2023-10-20 21:32:02 +00:00
lists.py [dynamo] Implement set.__contains__ for Tensor as object match of FakeTensor (#111738) 2023-10-22 17:40:34 +00:00
misc.py [dynamo] share a subgraph tracer across fwd and bwd in autograd.Function (#111588) 2023-10-20 21:32:02 +00:00
nn_module.py [dynamo] Expand _nonvar_fields names (#111749) 2023-10-23 02:58:16 +00:00
optimizer.py Fix typo under torch/_dynamo directory (#110459) 2023-10-04 16:05:05 +00:00
tensor.py dynamo: graph break on resize_ (#111553) 2023-10-22 02:27:14 +00:00
torch_function.py move tf override tensor to torch_function.py (#111714) 2023-10-21 02:29:01 +00:00
torch.py [dynamo] Fix context wrapping grad mode variable (#111534) 2023-10-22 20:55:48 +00:00
user_defined.py Support inference_mode decorator (#109274) 2023-09-27 22:21:42 +00:00