pytorch/torch/_dynamo/variables
2023-10-31 18:44:04 +00:00
..
__init__.py [5/N] Make torch context manager a TorchCtxManagerClassVariable (#111622) 2023-10-27 21:26:54 +00:00
base.py [dynamo] Remove VariableTracker.as_specialized (#112363) 2023-10-30 20:07:55 +00:00
builder.py [dynamo] Remove dead code - real_value_tensor_positive_aliases (#111911) 2023-10-30 23:10:52 +00:00
builtin.py [dynamo] Remove VariableTracker.as_specialized (#112363) 2023-10-30 20:07:55 +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] Replace recursively_contains with parents_tracker (#112122) 2023-10-28 06:46:48 +00:00
dicts.py Support 'BaseOutput' and subclasses from 'diffusers' in dynamo (#111978) 2023-10-30 19:53:31 +00:00
distributed.py [Easy] ConstantVariable() -> .create (#109896) 2023-09-22 22:30:15 +00:00
functions.py Support calling user defined triton kernels with kernel.run (#112292) 2023-10-30 17:51:23 +00:00
higher_order_ops.py [dynamo] Be stricter about HigherOrderOperator kwargs (#111938) 2023-10-28 18:54:33 +00:00
lists.py [dynamo] Replace recursively_contains with parents_tracker (#112122) 2023-10-28 06:46:48 +00:00
misc.py [dynamo] Remove mutation in AutogradFunctionContextVariable (#112216) 2023-10-28 06:46:48 +00:00
nn_module.py [dynamo] Expand _nonvar_fields names (#111749) 2023-10-23 02:58:16 +00:00
optimizer.py [dynamo] Replace recursively_contains with parents_tracker (#112122) 2023-10-28 06:46:48 +00:00
tensor.py [dynamo] Add support for register_post_accumulate_grad_hook (#112325) 2023-10-31 17:04:49 +00:00
torch_function.py Use pytree.arg_tree_leaves everywhere (#112394) 2023-10-31 15:57:06 +00:00
torch.py [dynamo] {*}Tensor.__init__ from list of Tensor/ndarray as torch.stack(List[FakeTensor]) (#111741) 2023-10-31 18:44:04 +00:00
user_defined.py [dynamo] fix None routing bug during var_getattr on UDO (#111614) 2023-10-29 01:57:43 +00:00