pytorch/torch/_export/db/examples
Zhengxu Chen e0d2a24967 Reland "[export] Support user input mutation. [1/2]" (#114496) (#114596)
Summary:

Serialization not implemented yet. Will do in the next diff.

Resolving Github issues:
https://github.com/pytorch/pytorch/issues/112429
https://github.com/pytorch/pytorch/issues/114142

Test Plan:
onnx doc test
```
python -m xdoctest /opt/conda/envs/py_3.8/lib/python3.8/site-packages/torch/onnx/_internal/exporter.py ONNXProgram.model_signature:0
```

Differential Revision: D51588558

Pull Request resolved: https://github.com/pytorch/pytorch/pull/114596
Approved by: https://github.com/angelayi
2023-11-27 20:19:04 +00:00
..
__init__.py
assume_constant_result.py
autograd_function.py
class_method.py
cond_branch_class_method.py
cond_branch_nested_function.py
cond_branch_nonlocal_variables.py
cond_closed_over_variable.py
cond_operands.py
cond_predicate.py
constrain_as_size_example.py Move export.constrain_as_* to torch._constrain_as_* (#110757) 2023-10-12 05:37:44 +00:00
constrain_as_value_example.py Move export.constrain_as_* to torch._constrain_as_* (#110757) 2023-10-12 05:37:44 +00:00
decorator.py
dictionary.py
dynamic_shape_assert.py
dynamic_shape_constructor.py
dynamic_shape_if_guard.py
dynamic_shape_map.py
dynamic_shape_round.py
dynamic_shape_slicing.py
dynamic_shape_view.py
fn_with_kwargs.py [BE]: ruff - enable PIE804 (#113951) 2023-11-17 21:23:02 +00:00
list_contains.py
list_unpack.py
model_attr_mutation.py [export oncall] add some examples during oncall (#112445) 2023-10-31 18:33:03 +00:00
nested_function.py
null_context_manager.py
optional_input.py [export oncall] add some examples during oncall (#112445) 2023-10-31 18:33:03 +00:00
pytree_flatten.py
scalar_output.py
specialized_attribute.py
static_for_loop.py
static_if.py
tensor_setattr.py
torch_sym_min.py [export oncall] add some examples during oncall (#112445) 2023-10-31 18:33:03 +00:00
type_reflection_method.py Uniformly use SourcelessBuilder to handle user defined types (#113390) 2023-11-10 07:26:52 +00:00
user_input_mutation.py Reland "[export] Support user input mutation. [1/2]" (#114496) (#114596) 2023-11-27 20:19:04 +00:00