pytorch/torch/_dynamo/variables
2023-08-21 19:07:20 +00:00
..
__init__.py dynamo: support disable_saved_tensors_hooks (#104869) 2023-07-26 07:27:37 +00:00
base.py [HigherOrderOp] Fall back on all new side effects in speculate_subgraph (#104077) 2023-06-28 14:20:37 +00:00
builder.py Guard numpy imports in the dynamo folder (#107299) 2023-08-21 19:07:20 +00:00
builtin.py Guard numpy imports in the dynamo folder (#107299) 2023-08-21 19:07:20 +00:00
constant.py NumPy support in torch.compile (#106211) 2023-08-11 00:39:32 +00:00
ctx_manager.py [dynamo] fix disable_saved_tensors_hooks - graph break (#106875) 2023-08-19 11:41:40 +00:00
dicts.py Better dynamo dict support via SetVariable keys (#106559) 2023-08-07 20:20:06 +00:00
distributed.py Grab bag of DTensor enablement stuff (Enable whole graph capture for DTensor) (#105787) 2023-07-30 00:17:45 +00:00
functions.py [dynamo] resolve InlinedClosureVariable in InstructionTranslator stack (#106491) 2023-08-03 16:45:42 +00:00
higher_order_ops.py [HigherOrderOp] stop erroring out on non-Tensor returns (#107461) 2023-08-21 13:39:36 +00:00
lists.py [BE]: Apply PYI autofixes to various types (#107521) 2023-08-20 02:42:21 +00:00
misc.py NumPy support in torch.compile (#106211) 2023-08-11 00:39:32 +00:00
nn_module.py Grab bag of DTensor enablement stuff (Enable whole graph capture for DTensor) (#105787) 2023-07-30 00:17:45 +00:00
optimizer.py Enable more e2e foreach optimizer compilation tests (#105438) 2023-07-20 02:41:19 +00:00
tensor.py Guard numpy imports in the dynamo folder (#107299) 2023-08-21 19:07:20 +00:00
torch.py Guard numpy imports in the dynamo folder (#107299) 2023-08-21 19:07:20 +00:00
user_defined.py [dynamo] Make KeyedJaggedTensor a variable. (#107319) 2023-08-18 17:15:46 +00:00