pytorch/torch/_dynamo/variables
2023-12-25 07:15:09 +00:00
..
__init__.py [Dynamo][10/N] Remove TorchVariable and is_allowed (#116312) 2023-12-23 09:44:09 +00:00
base.py Remove replace_all and make VTs mutable (#113725) 2023-12-10 09:31:21 +00:00
builder.py [Dynamo][11/N] allow_in_graph/disallow_in_graph decorator refactor (#116365) 2023-12-25 07:15:09 +00:00
builtin.py [Dynamo][10/N] Remove TorchVariable and is_allowed (#116312) 2023-12-23 09:44:09 +00:00
constant.py [dynamo] Improve support for dynamic shapes str.format and _assert (#115203) 2023-12-06 04:54:45 +00:00
ctx_manager.py [contextlib] Wrapping a function with set_grad_enabled will consume its global mutation (#113359) 2023-11-09 19:16:20 +00:00
dicts.py Remove replace_all and make VTs mutable (#113725) 2023-12-10 09:31:21 +00:00
distributed.py [DeviceMesh] Rename _device_mesh.py to device_mesh.py to prepare for beta (#115099) (#115193) 2023-12-08 08:44:32 +00:00
functions.py [inductor] make inductor work with new triton compile interface (#115878) 2023-12-22 00:09:29 +00:00
higher_order_ops.py [Dynamo][10/N] Remove TorchVariable and is_allowed (#116312) 2023-12-23 09:44:09 +00:00
iter.py [Dynamo][8/N] Wrap itertools.* as ItertoolsVariable (#115802) 2023-12-16 01:42:02 +00:00
lazy.py [dynamo] Remove VariableTracker.propagate (#111726) 2023-11-07 19:55:19 +00:00
lists.py Remove replace_all and make VTs mutable (#113725) 2023-12-10 09:31:21 +00:00
misc.py [Dynamo][11/N] allow_in_graph/disallow_in_graph decorator refactor (#116365) 2023-12-25 07:15:09 +00:00
nn_module.py [Dynamo][10/N] Remove TorchVariable and is_allowed (#116312) 2023-12-23 09:44:09 +00:00
optimizer.py Remove replace_all and make VTs mutable (#113725) 2023-12-10 09:31:21 +00:00
tensor.py [Dynamo][10/N] Remove TorchVariable and is_allowed (#116312) 2023-12-23 09:44:09 +00:00
torch_function.py Ensure wrapping subclasses with as_subclass is supported (#116091) 2023-12-20 14:37:08 +00:00
torch.py [Dynamo][11/N] allow_in_graph/disallow_in_graph decorator refactor (#116365) 2023-12-25 07:15:09 +00:00
user_defined.py [Dynamo][10/N] Remove TorchVariable and is_allowed (#116312) 2023-12-23 09:44:09 +00:00