Searched refs:tuple_to_list (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_library/ |
| H A D | infer_schema.py | 94 list_type = tuple_to_list(annotation_type)
255 def tuple_to_list(tuple_type: typing.Type[typing.Tuple]) -> typing.Type[typing.List]: function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_custom_ops.py | 22 from torch._library.infer_schema import tuple_to_list
3905 self.assertEqual(List, tuple_to_list(Tuple))
3909 result_type = tuple_to_list(Tuple[t, t, t])
3912 result_type = tuple_to_list(Tuple[t])
3917 result_type = tuple_to_list(Tuple[t, Optional[t]])
3920 result_type = tuple_to_list(Tuple[t, t, Optional[t]])
3923 result_type = tuple_to_list(Tuple[t, ...])
3927 result_type = tuple_to_list(Tuple[int, float])
3930 result_type = tuple_to_list(Tuple[int, float, str])
|