Searched defs:pre_dispatch (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | test_torchbind.py | 132 self, f, args, kwargs=None, strict=True, pre_dispatch=False argument
136 def export_wrapper(f, args, kwargs, strcit, pre_dispatch): argument
163 def test_none(self, pre_dispatch): argument
213 def test_attribute(self, pre_dispatch): argument
247 def test_attribute_as_custom_op_argument(self, pre_dispatch): argument
281 def test_input(self, pre_dispatch): argument
319 def test_input_as_custom_op_argument(self, pre_dispatch): argument
374 def test_torchbind_alias(self, pre_dispatch): argument
405 def test_torchbind_input_and_alias(self, pre_dispatch): argument
418 def test_unlift_custom_obj(self, pre_dispatch): argument
[all …]
|
| H A D | test_serialize.py | 553 def _check_graph(pre_dispatch): argument
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | utils.py | 117 def _has_potential_branch_input_mutation(branch, inputs, pre_dispatch=False): argument
157 def _has_potential_branch_input_alias(branch, inputs, pre_dispatch=False): argument
|
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | schemas.py | 820 pre_dispatch: bool = False variable in AOTConfig
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | functional_tensor.py | 315 def __init__(self, pre_dispatch=False, export=False, _allow_token_discovery=False): argument
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | _trace.py | 601 pre_dispatch=False, argument
|