| /aosp_15_r20/external/executorch/docs/source/tutorials_source/ |
| H A D | export-to-executorch-tutorial.py | 68 example_args = (torch.randn(1, 3, 256, 256),) variable
103 example_args = (torch.randn(3, 3), torch.randn(3, 3)) variable
130 example_args = (torch.randn(3, 3), torch.randn(3, 3)) variable
198 example_args = (torch.randn(1, 3, 256, 256),) variable
245 example_args = (torch.randn(1, 3, 256, 256),) variable
293 example_args = (torch.randn(1, 3, 256, 256),) variable
359 example_args = (torch.ones(1),) variable
425 example_args = (torch.ones(1),) variable
463 example_args = (torch.randn(2, 2), torch.randn(2, 2), torch.randn(2, 2)) variable
497 example_args = (torch.randn(2, 2), torch.randn(2, 2), torch.randn(2, 2)) variable
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_export/db/examples/ |
| H A D | dynamic_shape_slicing.py | 13 example_args = (torch.randn(3, 2),) variable
|
| H A D | list_unpack.py | 20 example_args = ([torch.randn(3, 2), torch.tensor(4), torch.tensor(5)],) variable
|
| H A D | static_for_loop.py | 15 example_args = (torch.randn(3, 2),) variable
|
| H A D | dynamic_shape_view.py | 15 example_args = (torch.randn(10, 10),) variable
|
| H A D | constrain_as_size_example.py | 20 example_args = (torch.tensor(4),) variable
|
| H A D | null_context_manager.py | 19 example_args = (torch.randn(3, 2),) variable
|
| H A D | static_if.py | 16 example_args = (torch.randn(3, 2, 2),) variable
|
| H A D | list_contains.py | 15 example_args = (torch.randn(3, 2),) variable
|
| H A D | user_input_mutation.py | 15 example_args = (torch.randn(3, 2),) variable
|
| H A D | dynamic_shape_constructor.py | 13 example_args = (torch.randn(3, 2),) variable
|
| H A D | dynamic_shape_assert.py | 16 example_args = (torch.randn(3, 2),) variable
|
| H A D | pytree_flatten.py | 15 example_args = ({1: torch.randn(3, 2), 2: torch.randn(3, 2)},), variable
|
| H A D | dynamic_shape_if_guard.py | 17 example_args = (torch.randn(3, 2, 2),) variable
|
| H A D | cond_predicate.py | 20 example_args = (torch.randn(6, 4, 3),) variable
|
| H A D | unsupported_operator.py | 15 example_args = (torch.randn(3, 2),) variable
|
| H A D | constrain_as_value_example.py | 23 example_args = (torch.tensor(4), torch.randn(5, 5)) variable
|
| H A D | dictionary.py | 15 example_args = (torch.randn(3, 2), torch.tensor(4)) variable
|
| H A D | tensor_setattr.py | 13 example_args = (torch.randn(3, 2), "attr") variable
|
| H A D | specialized_attribute.py | 25 example_args = (torch.randn(3, 2),) variable
|
| H A D | nested_function.py | 21 example_args = (torch.randn(3, 2), torch.randn(2)) variable
|
| H A D | optional_input.py | 17 example_args = (torch.randn(2, 3),) variable
|
| H A D | dynamic_shape_round.py | 17 example_args = (x,) variable
|
| H A D | dynamic_shape_map.py | 17 example_args = (torch.randn(3, 2), torch.randn(2)) variable
|
| H A D | class_method.py | 21 example_args = (torch.randn(3, 4),) variable
|